From c08ec86aaca58f3b1b9f4ce0e221866e69859328 Mon Sep 17 00:00:00 2001 From: pbellec Date: Wed, 1 Apr 2020 15:35:19 -0400 Subject: [PATCH] Fixes issue #46 --- dypac/bascpp.py | 2 +- dypac/dypac.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/dypac/bascpp.py b/dypac/bascpp.py index af4e06a..8cc67a4 100644 --- a/dypac/bascpp.py +++ b/dypac/bascpp.py @@ -135,7 +135,7 @@ def find_states(onehot, n_states=10, max_iter=30, threshold_sim=0.3, n_batch=0, """Find dynamic states based on the similarity of clusters over time.""" if verbose: print("Consensus clustering.") - cent, states, inert = k_means( + _, states, _ = k_means( onehot, n_clusters=n_states, init="k-means++", diff --git a/dypac/dypac.py b/dypac/dypac.py index 5914a85..99cbd82 100644 --- a/dypac/dypac.py +++ b/dypac/dypac.py @@ -293,7 +293,7 @@ def _mask_and_reduce_batch(self, imgs, confounds=None): init="k-means++", max_iter=self.max_iter, random_state=self.random_state, - n_init=self.n_init, + n_init=self.n_init_aggregation, ) # average stability maps and dwell times across consensus states @@ -342,7 +342,7 @@ def _mask_and_reduce(self, imgs, confounds=None): threshold_sim=self.threshold_sim, n_batch=self.n_batch, random_state=self.random_state, - n_init=self.n_init, + n_init=self.n_init_aggregation, verbose=self.verbose, )