From 341b4675735ca83cc2446c5e02d223e77b6dae51 Mon Sep 17 00:00:00 2001
From: Lizazal <66370193+Lizazal@users.noreply.github.com>
Date: Wed, 28 Sep 2022 13:52:42 +0300
Subject: [PATCH] Update classifier.py

---
 demos/classifier.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/demos/classifier.py b/demos/classifier.py
index 8d67d09f6..e671ab883 100755
--- a/demos/classifier.py
+++ b/demos/classifier.py
@@ -40,8 +40,8 @@
 from sklearn.discriminant_analysis import LinearDiscriminantAnalysis as LDA
 from sklearn.preprocessing import LabelEncoder
 from sklearn.svm import SVC
-from sklearn.grid_search import GridSearchCV
-from sklearn.mixture import GMM
+from sklearn.model_selection import GridSearchCV
+from sklearn import mixture
 from sklearn.tree import DecisionTreeClassifier
 from sklearn.naive_bayes import GaussianNB
 
@@ -131,7 +131,7 @@ def train(args):
         ]
         clf = GridSearchCV(SVC(C=1, probability=True), param_grid, cv=5)
     elif args.classifier == 'GMM':  # Doesn't work best
-        clf = GMM(n_components=nClasses)
+        clf = mixture.GaussianMixture(n_components=nClasses)
 
     # ref:
     # http://scikit-learn.org/stable/auto_examples/classification/plot_classifier_comparison.html#example-classification-plot-classifier-comparison-py