barclays_challenge_event_20.../models.py

10 lines
341 B
Python

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