try QDA

elleobrien · Aug 13, 2020 · 4a038ea · 4a038ea · github-actions · Aug 13, 2020
1 parent 0bac884
commit 4a038ea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -1,6 +1,6 @@
 import pandas as pd 
 import numpy as np
-from sklearn.linear_model import LogisticRegression
+from sklearn.discriminant_analysis import QuadraticDiscriminantAnalysis
 from sklearn import preprocessing
 from sklearn.model_selection import cross_val_predict
 from sklearn.metrics import confusion_matrix
@@ -28,7 +28,7 @@
 
 
 # Linear model
-clf = LogisticRegression()
+clf = QuadraticDiscriminantAnalysis()
 yhat = cross_val_predict(clf, X, y, cv=5)
 
 acc = np.mean(yhat==y)