diff --git a/build_model.py b/build_model.py index 85aa1f8..10a7b35 100644 --- a/build_model.py +++ b/build_model.py @@ -9,7 +9,7 @@ X_train, y_train = pickle.load(open(ROOT_DIR + "/data/german_train.p", "rb")) def buildRandomForestClassifier(): - creditClf = RandomForestClassifier(n_estimators=20, random_state=1) + creditClf = RandomForestClassifier(n_estimators=200, random_state=1) creditClf.fit(X=X_train, y=y_train) return creditClf @@ -18,4 +18,4 @@ def buildLogisticRegression(): creditClf.fit(X=X_train, y=y_train) return creditClf -build = buildRandomForestClassifier \ No newline at end of file +build = buildRandomForestClassifier diff --git a/build_model.pyc b/build_model.pyc new file mode 100644 index 0000000..d555743 Binary files /dev/null and b/build_model.pyc differ diff --git a/tests/__init__.pyc b/tests/__init__.pyc new file mode 100644 index 0000000..17e70a3 Binary files /dev/null and b/tests/__init__.pyc differ diff --git a/tests/test_build_model.pyc b/tests/test_build_model.pyc new file mode 100644 index 0000000..865ab33 Binary files /dev/null and b/tests/test_build_model.pyc differ