try QDA

.github/workflows/train.yaml

@@ -14,9 +14,10 @@ jobs:
           dvc repro 
 
           git fetch --prune
+          echo "## Metrics: Workspace vs. Main"
           dvc metrics diff --show-md master > report.md
 
           # Add figure to the report
-          echo "## Validating results by region"
+          echo "## Validating results by region" >> report.md
           cml-publish by_region.png --md >> report.md
-          cml-send-comment report.md
+          cml-send-comment report.md

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)