Skip to content

Instantly share code, notes, and snippets.

@rikturr

rikturr/rapids-random-forest.py

Created October 13, 2020 19:49
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save rikturr/b9694fe7a9f0dca8c494ca9e9bae5909 to your computer and use it in GitHub Desktop.
Save rikturr/b9694fe7a9f0dca8c494ca9e9bae5909 to your computer and use it in GitHub Desktop.
Download ZIP
rapids-random-forest
Raw
rapids-random-forest.py
from cuml.ensemble import RandomForestClassifier
# see notebook for prep_df function
taxi_train = prep_df(taxi)
rfc = RandomForestClassifier(n_estimators=100, max_depth=10, seed=42)
rfc.fit(taxi_train[features], taxi_train[y_col])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment