pep8 fixes in vis module
@@ -5,15 +5,13 @@ Definition of visualization methods
@author: baihan
"""
-import pyrsa as rsa
import numpy as np
-import matplotlib.pyplot as plt
from sklearn.manifold import MDS
-from sklearn.decomposition import PCA
sd = np.random.RandomState(seed=1)
mds_emb = MDS(n_components=2, random_state=sd, dissimilarity="precomputed")
+
def rdm_dimension_reduction(rdms, func):
""" dimension reduction of RDMs class
@@ -32,6 +30,7 @@ def rdm_dimension_reduction(rdms, func):
drs[i, :, :] = func.fit_transform(rdmm[i, :, :])
return drs
+
def mds(rdms):
""" multi-dimensional scaling of RDMs class
@@ -43,4 +42,3 @@ def mds(rdms):
"""
return rdm_dimension_reduction(rdms, mds_emb)
-
