Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from pyspark.ml.regression import RandomForestRegressionModel
- rf = RandomForestRegressor(labelCol="label",featuresCol="features", numTrees=5, maxDepth=10, seed=42)
- rf_model = rf.fit(train_df)
- rf_model_path = "./hdfsData/" + "rfr_model"
- rf_model.save(rf_model_path)
- rf_model.write.overwrite().save(rf_model_path)
- rf_model.write().overwrite().save(rf_model_path)
Add Comment
Please, Sign In to add comment