diff --git a/brainspace/gradient/gradient.py b/brainspace/gradient/gradient.py index d4765e7..260afad 100644 --- a/brainspace/gradient/gradient.py +++ b/brainspace/gradient/gradient.py @@ -84,7 +84,7 @@ class GradientMaps(BaseEstimator): Kernel function to build the affinity matrix. Possible options: {'pearson', 'spearman', 'cosine', 'normalized_angle', 'gaussian'}. If callable, must receive a 2D array and return a 2D square array. - If None, use input matrix. Default is None. + If None, use input matrix. Default is 'normalized_angle'. alignment : {'procrustes', 'joint'}, object or None Alignment approach. Only used when two or more datasets are provided. If None, no alignment is performed. If `object`, it accepts an instance @@ -110,7 +110,7 @@ class GradientMaps(BaseEstimator): is used. """ - def __init__(self, n_components=10, approach='dm', kernel=None, + def __init__(self, n_components=10, approach='dm', kernel='normalized_angle', alignment=None, random_state=None): self.n_components = n_components self.approach = approach diff --git a/requirements.txt b/requirements.txt index 77104f6..7eb3369 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ numpy>=1.11.0 scipy>=0.17.0 scikit-learn>=0.22.0 matplotlib>=2.0.0 -vtk>=8.1.0 +vtk>=9.1.0 nibabel nilearn