线性回归
import matplotlib.pyplot as plt
import numpy as np
from sklearn import linear_model
X = [[6], [8], [10], [14], [18]]
y = [[7], [9], [13], [17.5], [18]]
model=linear_model.LinearRegression()
model.fit(X,y)
print('预测一张12英寸匹萨价格:$%.2f' % model.predict(np.array([12]).reshape(-1,1))[0])
注:
- linear_model.LinearRegression()
- fit() #匹配
- predict