Skip to content

Commit 341b467

Browse files
Lizazalbamos
authored andcommitted
Update classifier.py
1 parent 0ac2558 commit 341b467

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

demos/classifier.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@
4040
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis as LDA
4141
from sklearn.preprocessing import LabelEncoder
4242
from sklearn.svm import SVC
43-
from sklearn.grid_search import GridSearchCV
44-
from sklearn.mixture import GMM
43+
from sklearn.model_selection import GridSearchCV
44+
from sklearn import mixture
4545
from sklearn.tree import DecisionTreeClassifier
4646
from sklearn.naive_bayes import GaussianNB
4747

@@ -131,7 +131,7 @@ def train(args):
131131
]
132132
clf = GridSearchCV(SVC(C=1, probability=True), param_grid, cv=5)
133133
elif args.classifier == 'GMM': # Doesn't work best
134-
clf = GMM(n_components=nClasses)
134+
clf = mixture.GaussianMixture(n_components=nClasses)
135135

136136
# ref:
137137
# http://scikit-learn.org/stable/auto_examples/classification/plot_classifier_comparison.html#example-classification-plot-classifier-comparison-py

0 commit comments

Comments
 (0)