Skip to content

Commit

Permalink
Merge pull request #773 from rasbt/pca-import
Browse files Browse the repository at this point in the history
fix pca import problem
  • Loading branch information
rasbt committed Feb 5, 2021
2 parents 894dce8 + 2af596e commit c79b08b
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions mlxtend/plotting/tests/test_pca_corr_graph.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,14 @@
import pytest
from mlxtend.plotting import plot_pca_correlation_graph
from mlxtend.data import iris_data
from sklearn.decomposition.pca import PCA

from distutils.version import LooseVersion as Version
from sklearn import __version__ as sklearn_version

if Version(sklearn_version) < '0.22':
from sklearn.decomposition.pca import PCA
else:
from sklearn.decomposition import PCA


def test_pass_pca_corr():
Expand Down Expand Up @@ -62,4 +69,3 @@ def test_not_enough_components():
plot_pca_correlation_graph(X, variables_names=['1', '2', '3', '4'],
X_pca=X_pca,
explained_variance=eigen[:-1])

0 comments on commit c79b08b

Please sign in to comment.