We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Artificial Intelligence
  3. Statistics and Machine Learning
  4. Correlation and Regression Lines - A Quick Recap #2
  5. Discussions

Correlation and Regression Lines - A Quick Recap #2

  • ericstricker73
     + 1 comment
    import numpy as np
from sklearn.linear_model import LinearRegression

Physics = [  15,  12,  8,   8,   7,   7,   7,   6,   5,   3]
History = [ 10,  25,  17,  11,  13,  17,  20,  13,  9,   15]

#Sample data
X = np.array(Physics).reshape((-1, 1))
y = np.array(History)

#Create and train the model
model = LinearRegression()
model.fit(X, y)

#Get the slope (coefficient)
slope = model.coef_[0]
slope