Upload New File

bb3b1ef2 · Rebecca Merrett · 3ce911fd · bb3b1ef2
Commit bb3b1ef2 authored Mar 07, 2019 by Rebecca Merrett
Show whitespace changes
Inline Side-by-side

Showing with 27 additions and 0 deletions

linear_regression.py Getting Started/linear_regression.py +27 -0

No files found.
--- a/Getting Started/linear_regression.py
+++ b/Getting Started/linear_regression.py
+from sklearn import linear_model
+from pandas import DataFrame
+import pandas as pd
+import pandas
+import matplotlib.pyplot as plt
+
+input_data = pandas.read_table("height.csv", header=0, sep=",", names=("weight", "height"))
+
+plt.scatter(input_data["weight"], input_data["height"])
+#plt.show()
+
+predictor = pd.DataFrame(input_data, columns=["weight"])
+outcome = pd.DataFrame(input_data, columns=["height"])
+
+lm = linear_model.LinearRegression()
+lm_model = lm.fit(predictor, outcome)
+
+predicted_heights = lm.predict(predictor)
+
+r_squared = lm.score(predictor,outcome)
+
+print(predicted_heights)
+#print("Predicted:")
+#print(predicted_heights[0:6])
+#print("Actual:")
+#print(outcome[0:6])
+#print(r_squared)