rikturr/rapids_rf.py

Created July 30, 2020 20:19

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/rikturr/edd617ea56f7051b4d5b65e72ef5d2ce.js"></script>
Save rikturr/edd617ea56f7051b4d5b65e72ef5d2ce to your computer and use it in GitHub Desktop.

Download ZIP

rapids_rf

Raw

rapids_rf.py

	from cuml.dask.ensemble import RandomForestRegressor
	rf = RandomForestRegressor(n_estimators=100, max_depth=10, seed=42)
	_ = rf.fit(X, y)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment