sathishleo commited on
Commit
cafb051
·
1 Parent(s): 6d489e7

Add app.py, backend, requirements, ignore models folder

Browse files
Files changed (1) hide show
  1. backend/train_model.py +2 -2
backend/train_model.py CHANGED
@@ -80,11 +80,11 @@ def train_model():
80
  models = {
81
  "LogReg_L1": Pipeline([
82
  ("scaler", StandardScaler()),
83
- ("clf", LogisticRegression(penalty="l1", solver="liblinear", max_iter=2000))
84
  ]),
85
  "LogReg_L2": Pipeline([
86
  ("scaler", StandardScaler()),
87
- ("clf", LogisticRegression(penalty="l2", solver="lbfgs", max_iter=2000))
88
  ]),
89
  "DecisionTree": DecisionTreeClassifier(random_state=42),
90
  "RandomForest": RandomForestClassifier(random_state=42),
 
80
  models = {
81
  "LogReg_L1": Pipeline([
82
  ("scaler", StandardScaler()),
83
+ ("clf", LogisticRegression(penalty="l1", solver="liblinear"))
84
  ]),
85
  "LogReg_L2": Pipeline([
86
  ("scaler", StandardScaler()),
87
+ ("clf", LogisticRegression(penalty="l2", solver="lbfgs"))
88
  ]),
89
  "DecisionTree": DecisionTreeClassifier(random_state=42),
90
  "RandomForest": RandomForestClassifier(random_state=42),