我正在尝试使用 ARIMA 模型来预测股票价格数据,具体来说,我正在使用 auto_arima。我的目标是预测未来 30 天的股价,并将其与测试数据进行比较。
这是我使用的代码:
#general
import pandas as pd
import numpy as np
import matplotlib.pylab as plt
%matplotlib inline
import yfinance as yf
ticker = "0118.KL"
data = yf.Ticker(ticker)
df = data.history(start="2019-01-01",end="2020-04-30")
df = df.filter(items=['Close'])
train = df[:-30]
test = df[-30:]
from pmdarima import auto_arima
model = auto_arima(train,trace=True,m=7,error_action='ignore', suppress_warnings=True)
model.fit(train)
forecast = model.predict(n_periods=30)
forecast = pd.DataFrame(forecast,index = test.index,columns=['Prediction'])
plt.plot(test, label='Valid')
plt.plot(forecast, label='Prediction')
plt.show()
关于如何更好地适应的任何想法?感谢您的阅读。