|
@@ -173,7 +173,7 @@ def main():
|
|
|
x_test_filters = self._data['x_test'].iloc[:, indices]
|
|
|
|
|
|
# model = _get_best_model(x_train_filters, y_train_filters)
|
|
|
- model = RandomForestClassifier(n_estimators=300, class_weight='balanced', n_jobs=-1)
|
|
|
+ model = RandomForestClassifier(n_estimators=500, class_weight='balanced', bootstrap=True, max_samples=0.75, n_jobs=-1)
|
|
|
model = model.fit(x_train_filters, y_train_filters)
|
|
|
|
|
|
y_test_model = model.predict(x_test_filters)
|