我有一个线性回归(显示polyfit,但只有1度保持线性),我试图返回从11个月到20个月的所有预测结果的数据集。目前我有10个月的数据和打印(速度)将打印结果在第20个月。
import numpy
from sklearn.metrics import r2_score
x = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
y = [1, 4, 3, 9, 8, 3, 7, 10, 8, 20]
mymodel = numpy.poly1d(numpy.polyfit(x, y, 1))
speed = mymodel(20)
print(speed)
预期结果(四舍五入)
import pandas as pd
df_results=pd.DataFrame({'months(x)':[11, 12, 13, 14, 15, 16, 17, 18, 19, 20], 'amount(y)':[15, 16, 17, 19, 20, 21, 23, 24, 26, 27]})
df_results
1条答案
按热度按时间j8ag8udp1#
如果我没理解错的话,你想得到第11个月到第20个月的预测。
这段代码应该会给予预期的结果。
我使用
numpy
round
函数对值进行四舍五入,并使用拟合模型得到预测。numpy
arange
函数创建从11日到20日的输入月份,步长为1。