barclays_challenge_event_20.../models.py

10 lines
341 B
Python
Raw Normal View History

2024-03-29 16:05:46 +00:00
from sklearn.ensemble import RandomForestClassifier
def random_forest_classifier(training_features, training_target):
model = RandomForestClassifier(max_features="log2",
random_state=79,
n_jobs=-1)
model.fit(training_features, training_target)
return model