File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 4040from sklearn .discriminant_analysis import LinearDiscriminantAnalysis as LDA
4141from sklearn .preprocessing import LabelEncoder
4242from 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
4545from sklearn .tree import DecisionTreeClassifier
4646from 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
You can’t perform that action at this time.
0 commit comments