Skip to content

Added Random Forest Regressor and tested with flake8 #1733

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Feb 7, 2020
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 42 additions & 0 deletions machine_learning/random_forest_regressor.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
# Random Forest Regressor Example

from sklearn.datasets import load_boston
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestRegressor
from sklearn.metrics import mean_absolute_error
from sklearn.metrics import mean_squared_error


def main():

"""
Random Tree Regressor Example using sklearn function.
Boston house price dataset is used to demonstrate algorithm.
"""

# Load Boston house price dataset
boston = load_boston()
print(boston.keys())

# Split dataset into train and test data
X = boston["data"] # features
Y = boston["target"]
x_train, x_test, y_train, y_test = train_test_split(
X, Y, test_size=0.3, random_state=1
)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

print(f"Mean Absolute Error(func) :\t {mean_absolute_error(y_test, predictions)}")
print(f"Mean Square Error(func) :\t {mean_squared_error(y_test, predictions)}")

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

OK, on it!


# Random Forest Regressor
rand_for = RandomForestRegressor(random_state=42, n_estimators=300)
rand_for.fit(x_train, y_train)

# Predict target for test data
predictions = rand_for.predict(x_test)
predictions = predictions.reshape(len(predictions), 1)

# Error printing
print(f"Mean Absolute Error:\t {mean_absolute_error(y_test, predictions)}")
print(f"Mean Square Error :\t {mean_squared_error(y_test, predictions)}")


if __name__ == "__main__":
main()