analyser_hj3415 2.7.7__py2.py3-none-any.whl → 2.7.8__py2.py3-none-any.whl

analyser_hj3415/myprophet.py

@@ -0,0 +1,144 @@
+import yfinance as yf
+from datetime import datetime, timedelta
+import pandas as pd
+from prophet import Prophet
+from sklearn.preprocessing import StandardScaler
+from utils_hj3415 import utils
+from typing import Optional
+import plotly.graph_objs as go
+from plotly.offline import plot
+import matplotlib.pyplot as plt  # Matplotlib 수동 임포트
+from db_hj3415 import myredis
+
+class MyProphet:
+    def __init__(self, code: str):
+        assert utils.is_6digit(code), f'Invalid value : {code}'
+        self._code = code
+        self.name = myredis.Corps(code, 'c101').get_name()
+        self.raw_data = self.get_raw_data()
+
+        self.scaler = StandardScaler()
+        self.model = Prophet()
+
+    @property
+    def code(self) -> str:
+        return self._code
+
+    @code.setter
+    def code(self, code: str):
+        assert utils.is_6digit(code), f'Invalid value : {code}'
+        self._code = code
+        self.name = myredis.Corps(code, 'c101').get_name()
+        self.raw_data = self.get_raw_data()
+
+    def get_raw_data(self) -> pd.DataFrame:
+        """
+        야후에서 해당 종목의 4년간 주가 raw data를 받아온다.
+        :return:
+        """
+        # 오늘 날짜 가져오기
+        today = datetime.today()
+
+        # 4년 전 날짜 계산 (4년 = 365일 * 4)
+        four_years_ago = today - timedelta(days=365 * 4)
+
+        return yf.download(
+            self.code + '.KS',
+            start=four_years_ago.strftime('%Y-%m-%d'),
+            end=today.strftime('%Y-%m-%d')
+        )
+
+    def preprocessing_for_prophet(self) -> pd.DataFrame:
+        """
+        Prophet이 사용할 수 있도록 데이터 준비
+        ds는 날짜, y는 주가
+        :return:
+        """
+        df = self.raw_data[['Close', 'Volume']].reset_index()
+        df.columns = ['ds', 'y', 'volume']  # Prophet의 형식에 맞게 열 이름 변경
+
+        # 추가 변수를 정규화
+        df['volume_scaled'] = self.scaler.fit_transform(df[['volume']])
+        return df
+
+    def make_forecast(self) -> pd.DataFrame:
+        # Prophet을 위한 dataframe 만들기
+        df_for_prophet = self.preprocessing_for_prophet()
+
+        # 정규화된 'volume_scaled' 변수를 외부 변수로 추가
+        self.model.add_regressor('volume_scaled')
+
+        self.model.fit(df_for_prophet)
+
+        # 향후 180일 동안의 주가 예측
+        future = self.model.make_future_dataframe(periods=180)
+
+        # 미래 데이터에 거래량 추가 (평균 거래량을 사용해 정규화)
+        future_volume = pd.DataFrame({'volume': [self.raw_data['Volume'].mean()] * len(future)})
+        future['volume_scaled'] = self.scaler.transform(future_volume[['volume']])
+
+        forecast = self.model.predict(future)
+        return forecast
+
+    def export_to(self, to="str") -> Optional[str]:
+        """
+        prophet과 plotly로 그래프를 그려서 html을 문자열로 반환
+        :param to: str, png, htmlfile, show
+        :return:
+        """
+        # 실제 데이터
+        df = self.preprocessing_for_prophet()
+        # 예측 데이터
+        forecast = self.make_forecast()
+
+        # Plotly를 사용한 시각화
+        fig = go.Figure()
+
+        # 실제 데이터
+        fig.add_trace(go.Scatter(x=df['ds'], y=df['y'], mode='markers', name='실제주가'))
+        # 예측 데이터
+        fig.add_trace(go.Scatter(x=forecast['ds'], y=forecast['yhat'], mode='lines', name='예측치'))
+
+        # 상한/하한 구간
+        fig.add_trace(
+            go.Scatter(x=forecast['ds'], y=forecast['yhat_upper'], fill=None, mode='lines', name='상한'))
+        fig.add_trace(
+            go.Scatter(x=forecast['ds'], y=forecast['yhat_lower'], fill='tonexty', mode='lines', name='하한'))
+
+        fig.update_layout(
+            title=f'{self.code} {self.name} 주가 예측 그래프(prophet)',
+            xaxis_title='일자',
+            yaxis_title='주가(원)',
+            xaxis = dict(
+                tickformat='%Y/%m',  # X축을 '연/월' 형식으로 표시
+            ),
+            yaxis = dict(
+                tickformat=".0f",  # 소수점 없이 원래 숫자 표시
+            )
+        )
+
+        if to == 'str':
+            # 그래프 HTML로 변환 (string 형식으로 저장)
+            graph_html = plot(fig, output_type='div')
+            return graph_html
+        elif to == 'png':
+            # 그래프를 PNG 파일로 저장
+            fig.write_image("plotly_graph.png")
+        elif to == 'htmlfile':
+            # 그래프를 HTML로 저장
+            plot(fig, filename='graph_plotly.html', auto_open=False)
+            return None
+        elif to == 'show':
+            # 예측 결과 출력
+            print(forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail())
+            # 예측 결과 시각화 (Matplotlib 사용)
+            fig = self.model.plot(forecast)
+            # 추세 및 계절성 시각화
+            fig2 = self.model.plot_components(forecast)
+            plt.show()  # 시각화 창 띄우기
+        else:
+            Exception("to 인자가 맞지 않습니다.")
+
+
+
+

analyser_hj3415/workroom/lstm.py

@@ -9,11 +9,11 @@ import matplotlib.pyplot as plt
 # 1. 데이터 다운로드 (애플 주식 데이터를 사용)
 #stock_data = yf.download('AAPL', start='2020-01-01', end='2023-01-01')
 # 삼성전자 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('005930.KS', start='2019-01-01', end='2024-10-10')
+#stock_data = yf.download('005930.KS', start='2019-01-01', end='2024-10-11')
 # 크래프톤 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('259960.KS', start='2020-01-01', end='2024-10-08')
+stock_data = yf.download('259960.KS', start='2020-01-01', end='2024-10-11')
 # 하이닉스 주식 데이터 가져오기 (KOSPI 상장)
-stock_data = yf.download('000660.KS', start='2019-01-01', end='2024-10-10')
+#stock_data = yf.download('000660.KS', start='2019-01-01', end='2024-10-11')
 
 
 # 2. 필요한 열만 선택 (종가만 사용)

{analyser_hj3415-2.7.7.dist-info → analyser_hj3415-2.7.8.dist-info}/METADATA

@@ -1,12 +1,18 @@
 Metadata-Version: 2.1
 Name: analyser_hj3415
-Version: 2.7.7
+Version: 2.7.8
 Summary: Stock analyser and database processing programs
 Author-email: Hyungjin Kim <hj3415@gmail.com>
 Description-Content-Type: text/markdown
 Classifier: License :: OSI Approved :: MIT License
 Requires-Dist: utils-hj3415>=2.9.2
 Requires-Dist: db-hj3415>=4.0.3
+Requires-Dist: scikit-learn>=1.5.2
+Requires-Dist: plotly>=5.24.1
+Requires-Dist: yfinance>=0.2.44
+Requires-Dist: prophet>=1.1.6
+Requires-Dist: kaleido>=0.2.1
+Requires-Dist: matplotlib>=3.9.2
 Project-URL: Home, https://www.hyungjin.kr
 
 ### analyser-hj3415

{analyser_hj3415-2.7.7.dist-info → analyser_hj3415-2.7.8.dist-info}/RECORD

@@ -2,15 +2,15 @@ analyser_hj3415/.DS_Store,sha256=qr9-0FPn5CFKe6kEu8_dWCNhzQ0sN7bwQgffKsaJEEo,614
 analyser_hj3415/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 analyser_hj3415/cli.py,sha256=7ys-b9H_tJAbLLi_XpjmEvDiyNHddpg9F1FNAACfWnE,11005
 analyser_hj3415/eval.py,sha256=4F0GIknCogAhv_iTq8auLrmwW20u8kH0HY0fP4SaVa4,39099
+analyser_hj3415/myprophet.py,sha256=N_eUzAmrbEuqJBJ9OqbnMrHC3_kc3JXGq6pND5o4yuE,5096
 analyser_hj3415/workroom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-analyser_hj3415/workroom/lstm.py,sha256=O_VYURGNsLw6q_7Yi0nH4Y6nKbVBwzMojDN446cyJKM,4078
-analyser_hj3415/workroom/myprophet.py,sha256=xGbs4_cCeSNAX62H6rG-bUz6_rUOyO6787WAjLexMkw,1855
+analyser_hj3415/workroom/lstm.py,sha256=gF7i1NqPOhNysvkSQcC_mkOmvLk96jPxoqajZ-FFD8I,4078
 analyser_hj3415/workroom/mysklearn.py,sha256=wJXKz5MqqTzADdG2mqRMMzc_G9RzwYjj5_j4gyOopxQ,2030
 analyser_hj3415/workroom/mysklearn2.py,sha256=1lIy6EWEQHkOzDS-av8U0zQH6DuCLKWMI73dnJx5KRs,1495
 analyser_hj3415/workroom/score.py,sha256=P6nHBJYmyhigGtT4qna4BmNtvt4B93b7SKyzdstJK24,17376
 analyser_hj3415/workroom/trash.py,sha256=zF-W0piqkGr66UP6-iybo9EXh2gO0RP6R1FnIpsGkl8,12262
-analyser_hj3415-2.7.7.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
-analyser_hj3415-2.7.7.dist-info/LICENSE,sha256=QVKTp0dTnB5xG8RLgG17LwSWCKNEzYoVVM6KjoCPKc0,1079
-analyser_hj3415-2.7.7.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
-analyser_hj3415-2.7.7.dist-info/METADATA,sha256=KGOyS4fsSdfpqpHu8O6O2YewMxWQGi1W8Sj45ixtYkk,6417
-analyser_hj3415-2.7.7.dist-info/RECORD,,
+analyser_hj3415-2.7.8.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
+analyser_hj3415-2.7.8.dist-info/LICENSE,sha256=QVKTp0dTnB5xG8RLgG17LwSWCKNEzYoVVM6KjoCPKc0,1079
+analyser_hj3415-2.7.8.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
+analyser_hj3415-2.7.8.dist-info/METADATA,sha256=Ru0TFe19ItTPkffO_39zz7vuX0b78V7EZoiO4f1wCEM,6607
+analyser_hj3415-2.7.8.dist-info/RECORD,,

analyser_hj3415/workroom/myprophet.py

@@ -1,58 +0,0 @@
-# prophet 사용 공부
-
-import yfinance as yf
-import pandas as pd
-from prophet import Prophet
-import matplotlib.pyplot as plt  # Matplotlib 수동 임포트
-
-
-# 애플 주식 데이터를 다운로드
-#stock_data = yf.download('AAPL', start='2020-01-01', end='2024-10-09')
-# 삼성전자 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('005930.KS', start='2020-01-01', end='2024-08-01')
-# 크래프톤 주식 데이터 가져오기 (KOSPI 상장)
-#stock_data = yf.download('259960.KS', start='2020-01-01', end='2024-10-08')
-# 하이닉스 주식 데이터 가져오기 (KOSPI 상장)
-stock_data = yf.download('000660.KS', start='2020-01-01', end='2024-10-08')
-
-
-# Prophet이 사용할 수 있도록 데이터 준비
-df = stock_data[['Close', 'Volume']].reset_index()
-df.columns = ['ds', 'y', 'volume']  # Prophet의 형식에 맞게 열 이름 변경
-
-from sklearn.preprocessing import StandardScaler
-
-# 추가 변수를 정규화
-scaler = StandardScaler()
-df['volume_scaled'] = scaler.fit_transform(df[['volume']])
-
-# Prophet 모델 생성 및 학습
-model = Prophet()
-
-# 정규화된 'volume_scaled' 변수를 외부 변수로 추가
-model.add_regressor('volume_scaled')
-
-model.fit(df)
-
-# 향후 180일 동안의 주가 예측
-future = model.make_future_dataframe(periods=180)
-
-# 미래 데이터에 거래량 추가 (평균 거래량을 사용해 정규화)
-future_volume = pd.DataFrame({'volume': [stock_data['Volume'].mean()] * len(future)})
-future['volume_scaled'] = scaler.transform(future_volume[['volume']])
-
-
-forecast = model.predict(future)
-
-
-
-# 예측 결과 출력
-print(forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail())
-
-# 예측 결과 시각화 (Matplotlib 사용)
-fig = model.plot(forecast)
-
-# 추세 및 계절성 시각화
-fig2 = model.plot_components(forecast)
-
-plt.show()  # 시각화 창 띄우기