rikturr/spark_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/1b30990dd5ce6d742699793cd11cbf76.js"></script>
Save rikturr/1b30990dd5ce6d742699793cd11cbf76 to your computer and use it in GitHub Desktop.

Download ZIP

spark_rf

Raw

spark_rf.py

	from pyspark.ml.regression import RandomForestRegressor
	rf = RandomForestRegressor(numTrees=100, maxDepth=10, seed=42)
	fitted = rf.fit(X)

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