Spaces:
Sleeping
Sleeping
Commit
·
cafb051
1
Parent(s):
6d489e7
Add app.py, backend, requirements, ignore models folder
Browse files- 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"
|
| 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),
|
|
|
|
| 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),
|