Advertisement
Guest User

Untitled

a guest
Apr 26th, 2019
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.45 KB | None | 0 0
  1. #define a function for random forest regressor
  2. from sklearn.ensemble import RandomForestRegressor
  3. def RandomForestRegressorModel(X,y):
  4. X_train, X_test, y_train, y_test = train_test_split(X, y,random_state=42, test_size=0.3)
  5. rf = RandomForestRegressor(random_state=42)
  6. rf.fit(X_train, y_train)
  7. y_pred = rf.predict(X_test)
  8. print(print_accuracy_report(y_test, y_pred, X_test, rf))
  9. return rf
  10. randomForestModel = RandomForestRegressorModel(X,y)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement