Skip to content
2 changes: 1 addition & 1 deletion pydfc/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -361,9 +361,9 @@ def nifti2array(

return time_series, labels, locs


def nifti2timeseries(
nifti_file,
n_rois,
Fs,
subj_id,
confound_strategy="none",
Expand Down
5 changes: 5 additions & 0 deletions pydfc/dfc_methods/cap.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
import numpy as np
from scipy.special import softmax
from sklearn.cluster import KMeans
from sklearn.metrics import silhouette_samples

from ..dfc import DFC
from ..time_series import TIME_SERIES
Expand Down Expand Up @@ -38,6 +39,7 @@ class CAP(BaseDFCMethod):
def __init__(self, **params):
self.logs_ = ""
self.FCS_ = []
self.silhouette_sc_ = []
self.mean_act = []
self.FCS_fit_time_ = None
self.dFC_assess_time_ = None
Expand Down Expand Up @@ -126,6 +128,9 @@ def estimate_FCS(self, time_series):
self.FCS_ = self.act_vec2FCS(group_act_centroids)
self.Z = self.kmeans_.predict(time_series.data.T.astype(np.float32))

# silhouette coefficient
self.silhouette_sc_ = silhouette_samples(time_series.data.T.astype(np.float32),self.Z)

# mean activation of states
self.set_mean_activity(time_series)

Expand Down