Added required PCA import for cvPCA

MouseLand · Oct 23, 2023 · 4f8489d · 4f8489d
1 parent e92a7e0
commit 4f8489d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/neuropop/dimensionality.py b/neuropop/dimensionality.py
@@ -124,6 +124,7 @@ def scvPCA(A, B):
 
 def cvPCA(X):
     ''' X is 2 x stimuli x neurons '''
+    from sklearn.decomposition import PCA
     nr = X.shape[0]
     pca = PCA(n_components=min(1024, X.shape[1])).fit(X[0])
     #u = pca.components_.T