采用scikit-learn 库推算一个简单的线性回归

假如 小明 2015 年 4 岁，2016 年 5 岁，2017 年 6 岁，那 2020 年小明几岁？
虽然这个例子非常简单，但是也可以看成一个简单线性回归问题

from sklearn import linear_model

model = linear_model.LinearRegression()

model.fit([[2015],[2016],[2017],[2018]],[4,5,6,7])

result = model.predict([[2020]])

print(result)

返回结果

[9.]