Skip to content

Instantly share code, notes, and snippets.

@glamp

glamp/scikit-example.py

Created May 22, 2019 16:43
Show Gist options
  • Save glamp/bace35ff7e5a0008abb34e6c190abc28 to your computer and use it in GitHub Desktop.
Save glamp/bace35ff7e5a0008abb34e6c190abc28 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
scikit-example.py
from sklearn.ensemble import RandomForestClassifier
clf = RandomForestClassifier()
target_variable = 'does-make-more-than-50k'
columns = ['age', 'education', 'hours-worked-per-week']
clf.fit(df[columns], df[target_variable])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment