++ timezone 오류 수정완료
https://colab.research.google.com/drive/1RnWMRf5nx9lSLWm9-h33BHD0HFN6r6Lq?usp=sharing
파이썬의 pykrx와 prophet 모듈을 이용한 주식 예측 코드입니다.
1 | !pip install pykrx | cs |
1 2 3 | from pykrx import stock from fbprophet import Prophet from fbprophet.plot import add_changepoints_to_plot | cs |
1 2 3 4 | # 종목, 기간 선택 dd = stock.get_market_ohlcv(“20210101”, “20220610”, “051600”) df = dd.tz_localize(None) | cs |
++ timezone 오류 수정완료했습니다.
종목은 코드로 입력하시면 되고 기간은 수동으로 코드를 실행하는 당일 날짜까지 설정해야합니다.
예시의 코드는 한전 kps의 코드입니다.
1 2 3 4 5 | # 시간(ds)와 종가(y)값만 남김 df = df.reset_index() df[‘ds’] = df[‘날짜’] df[‘y’] = df[‘종가’] data = df[[‘ds’, ‘y’]] | cs |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 | #학습 #튜닝 전 #model = Prophet() #model.fit(data) #학습 #튜닝 후 model = Prophet( seasonality_mode=‘multiplicative’, changepoint_prior_scale=0.5, changepoint_range=0.9, ) model.fit(data) | cs |
1 2 3 4 | #1개월 예측 future = model.make_future_dataframe(periods=1, freq=‘M’) forecast = model.predict(future) | cs |
1 2 3 | # 그래프1 fig1 = model.plot(forecast) a = add_changepoints_to_plot(fig1.gca(), model, forecast) | cs |
1 2 | # 그래프2 fig2 = model.plot_components(forecast) | cs |
1 | forecast | cs |
이 코드들을 복사하셔서 사용하셔도 되고 상단의 구글 Colab 링크로 바로 들어가셔서 사용하셔도 됩니다.