sathishleo commited on
Commit
211a2cd
·
1 Parent(s): 3182f0c

Add app.py, backend, and model for HF Space

Browse files
Files changed (1) hide show
  1. backend/train_model.py +4 -3
backend/train_model.py CHANGED
@@ -19,11 +19,12 @@ from sklearn.metrics import (
19
  from sklearn.linear_model import LogisticRegression
20
  from sklearn.tree import DecisionTreeClassifier
21
  from sklearn.ensemble import RandomForestClassifier, BaggingClassifier
22
- from utils import (
23
- ensure_dirs, save_json, plot_cm, plot_roc, barplot_metric,
24
- lineplot_curves
25
  )
26
 
 
27
  # ------------------------------
28
  # Paths
29
  # ------------------------------
 
19
  from sklearn.linear_model import LogisticRegression
20
  from sklearn.tree import DecisionTreeClassifier
21
  from sklearn.ensemble import RandomForestClassifier, BaggingClassifier
22
+ from .utils import (
23
+ ensure_dirs, save_json, plot_cm, plot_roc, barplot_metric,
24
+ lineplot_curves
25
  )
26
 
27
+
28
  # ------------------------------
29
  # Paths
30
  # ------------------------------