From 9591b818ede1b4f8571a783cd8b93c2464e8ee34 Mon Sep 17 00:00:00 2001 From: esavary Date: Thu, 28 Mar 2024 15:00:10 +0100 Subject: [PATCH] Update src/eddymotion/estimator.py Co-authored-by: Oscar Esteban --- src/eddymotion/estimator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/eddymotion/estimator.py b/src/eddymotion/estimator.py index 79852618..1cafb233 100644 --- a/src/eddymotion/estimator.py +++ b/src/eddymotion/estimator.py @@ -151,7 +151,7 @@ def fit( pbar.set_description_str( f"Pass {i_iter + 1}/{n_iter} | Fit and predict b-index <{i}>" ) - data_train, data_test = lovo_split(dwdata, i) + data_train, data_test = lovo_split(dwdata, i, with_b0=True) grad_str = f"{i}, {data_test[1][:3]}, b={int(data_test[1][3])}" pbar.set_description_str(f"[{grad_str}], {n_jobs} jobs")