[Python] 간단한 딥러닝 코인 예측 프로그램 만들기

pc에 설치한 파이썬을 바로 이용하기보단 Google의 Colab을 사용하시는게 더 편하실 겁니다.

1
2
!pip install pyupbit
!pip install pykrx
cs
1
2
3
4
import pyupbit
from pykrx import stock
from fbprophet import Prophet
from fbprophet.plot import add_changepoints_to_plot
cs
  1. (기존 출처의 코드에서 timezone 오류 수정완료)
1
2
3
4
# BTC 최근 200시간의 데이터 불러옴
 
dd = pyupbit.get_ohlcv(“KRW-BTC”, interval=“minute60”)
df = dd.tz_localize(None)
cs
1
2
3
4
5
# 시간(ds)와 종가(y)값만 남김
df = df.reset_index()
df[‘ds’= df[‘index’]
df[‘y’= df[‘close’]
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
#24시간 예측
future = model.make_future_dataframe(periods=24, freq=‘H’)
 
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
2
3
4
5
6
#가격 조회
nowValue = pyupbit.get_current_price(“KRW-BTC”)
closeValue = forecast[‘yhat’].values[1]
 
print(“현재가 : “ , nowValue)
print(“24시간뒤의 가격: “, closeValue)
cs
1
forecast
cs

조금씩 조정해가면서 조절하면 참고할 정도는 되는것 같습니다.

“출처 : https://github.com/Maker-Kim/Study/blob/master/python/Coin_Ai.ipynb

++ 최근 출처 내의 코드 중에서 그대로 colab에서 실행하면 timezone 오류가 발생하게 되어 코드를 한부분 수정했습니다.

https://colab.research.google.com/drive/1mnsqaE1AowwlQ9456kpBAOL7ranBVP0J?usp=sharing

위 링크가 수정한 코드입니다.

관련글

in-coder 커뮤니티 HOT게시물

제목 작성자 작성일