Merge pull request #18 from MLOPsStudyGroup/RandomForestClassifier

arthurolga · web-flow · commit 4e46244047e7 · 2021-05-19T17:06:08.000-03:00
Random Forest Experiment
diff --git a/src/model.py b/src/model.py
@@ -43,7 +43,15 @@ def train(data, num_estimators, isDataFrame=False):
     pipe = Pipeline(
         [
             ("scaler", StandardScaler()),
-            ("LR", LogisticRegression(random_state=0, max_iter=num_estimators)),
+            (
+                "RFC",
+                RandomForestClassifier(
+                    criterion="gini",
+                    max_depth=10,
+                    max_features="auto",
+                    n_estimators=num_estimators,
+                ),
+            ),
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,15 @@ def train(data, num_estimators, isDataFrame=False):`
`43`	`43`	`pipe = Pipeline(`
`44`	`44`	`[`
`45`	`45`	`("scaler", StandardScaler()),`
`46`		`- ("LR", LogisticRegression(random_state=0, max_iter=num_estimators)),`
	`46`	`+ (`
	`47`	`+ "RFC",`
	`48`	`+ RandomForestClassifier(`
	`49`	`+ criterion="gini",`
	`50`	`+ max_depth=10,`
	`51`	`+ max_features="auto",`
	`52`	`+ n_estimators=num_estimators,`
	`53`	`+ ),`
	`54`	`+ ),`
`47`	`55`	`]`
`48`	`56`	`)`
`49`	`57`