Skip to content

Instantly share code, notes, and snippets.

@vivekpadia70

vivekpadia70/house_prediction.py

Created December 25, 2019 06:20
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 vivekpadia70/1dee9be8a4f34443eaf8bfef9f2100de to your computer and use it in GitHub Desktop.
Save vivekpadia70/1dee9be8a4f34443eaf8bfef9f2100de to your computer and use it in GitHub Desktop.
Download ZIP
Raw
house_prediction.py
rfr = RandomForestRegressor(n_estimators=1000, max_depth=5, n_jobs=-1, random_state=12)
rfr.fit(X_train, Y_train)
print("Random Forest R^2 Score: ", rfr.score(X_train, Y_train))
print("Random Forest Test R^2 Score: ", rfr.score(X_test, Y_test))
y_pred = rfr.predict(X_test)
print("Mean Squared Error: ", mean_squared_error(y_pred, Y_test))
print("Mean Absolute Error: ", mean_absolute_error(y_pred, Y_test))
print("Cross Validation Score: ", cross_val_score(rfr, X_test, Y_test, cv=5))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment