Skip to content

Instantly share code, notes, and snippets.

@WillKoehrsen

WillKoehrsen/random_forest.py

Last active July 12, 2021 13:38
Show Gist options
  • Save WillKoehrsen/009862ad00ba7f3767c904506a7d8a54 to your computer and use it in GitHub Desktop.
Save WillKoehrsen/009862ad00ba7f3767c904506a7d8a54 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
random_forest.py
from sklearn.ensemble import RandomForestClassifier
# Create the model with 100 trees
model = RandomForestClassifier(n_estimators=100,
bootstrap = True,
max_features = 'sqrt')
# Fit on training data
model.fit(train, train_labels)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment