Very basic single linear regression example in Python

single-linear-regression.py

# Very basic single linear regression example in Python. Daniel Kreider Oct. 28 2017
# 
#
#  X(input)
#  9 |
#  8 |                *
#  7 |              *
#  6 |            *
#  5 |          *
#  4 |        *
#  3 |      *
#  2 |    * 
#  1 |_ _ _ _ _ _ _ _ _ 
#     1 2 3 4 5 6 7 8 9 Y(output)

import pandas as pd
from sklearn.linear_model import LinearRegression
from matplotlib import pyplot as plt

XData = [1,2,3,4,5,6,7,8,9]; # Input
yData = [2,3,4,5,6,7,8,9,10]; # Outcome

X = pd.DataFrame(XData);
y = pd.DataFrame(yData);

lm = LinearRegression();
model = lm.fit(X, y);

xTest = input("Please enter a number: ");
prediction = model.predict(int(xTest));
print("Based on your input, the trained model returned: ", prediction[0]);