analyser_hj3415 2.9.18__py3-none-any.whl → 2.10.1__py3-none-any.whl

analyser_hj3415/tsa.py

@@ -192,6 +192,21 @@ class MyProphet:
         else:
             Exception("to 인자가 맞지 않습니다.")
 
+    def scoring(self) -> int:
+        last_real_data = self.df_real.iloc[-1]
+        recent_price = last_real_data['y']
+        recent_date = datetime.strftime(last_real_data['ds'], '%Y-%m-%d')
+        yhat_dict = self.get_yhat()
+        tsa_logger.info(f'recent_price: {recent_price}, yhat_dict: {yhat_dict}')
+        yhat_lower = int(yhat_dict['yhat_lower'])
+        deviation = int(eval.Tools.cal_deviation(recent_price, yhat_lower))
+        if recent_price > yhat_lower:
+            score = -deviation
+        else:
+            score = deviation
+        print(f"{self.code}/{self.name} date: {recent_date} 가격:{recent_price} 기대하한값:{yhat_lower} 편차:{deviation} score:{score}")
+        return score
+
     @classmethod
     def ranking(cls, refresh = False, expire_time_h = 24, top='all') -> OrderedDict:
         """
@@ -208,22 +223,14 @@ class MyProphet:
         def fetch_ranking() -> dict:
             data = {}
             p = MyProphet('005930')
-            for i, code in enumerate(myredis.Corps.list_all_codes()):
+            for code in myredis.Corps.list_all_codes():
                 try:
                     p.code = code
                 except ValueError:
                     tsa_logger.error(f'myprophet ranking error : {code}/{myredis.Corps(code, "c101").get_name()}')
                     continue
-                last_real_data = p._preprocessing_for_prophet().iloc[-1]
-                recent_price = last_real_data['y']
-                recent_date = datetime.strftime(last_real_data['ds'], '%Y-%m-%d')
-                yhat_dict = p.get_yhat()
-                tsa_logger.info(f'recent_price: {recent_price}, yhat_dict: {yhat_dict}')
-                yhat_lower = int(yhat_dict['yhat_lower'])
-                if recent_price < yhat_lower:
-                    deviation = int(eval.Tools.cal_deviation(recent_price, yhat_lower))
-                    data[code] = deviation
-                    print(f"{i}.{p.code}/{p.name} date: {recent_date} 가격:{recent_price} 기대하한값:{yhat_lower} 편차:{deviation}")
+                score = p.scoring()
+                data[code] = score
             return data
 
         data_dict = myredis.Base.fetch_and_cache_data(redis_name, refresh, fetch_ranking, timer=expire_time_h * 3600)
@@ -346,12 +353,28 @@ class MyLSTM:
                 y.append(data[i + time_step, 0])
             return np.array(X), np.array(y)
 
+        
         X_train, y_train_1d = create_dataset(train_data_2d)
         X_test, y_test_1d = create_dataset(test_data_2d)
+        tsa_logger.debug(X_train.shape)
+        tsa_logger.debug(X_test.shape)
 
-        # LSTM 모델 입력을 위해 데이터를 3차원으로 변환
-        X_train_3d = X_train.reshape(X_train.shape[0], X_train.shape[1], 1)
-        X_test_3d = X_test.reshape(X_test.shape[0], X_test.shape[1], 1)
+        try:
+            # LSTM 모델 입력을 위해 데이터를 3차원으로 변환
+            X_train_3d = X_train.reshape(X_train.shape[0], X_train.shape[1], 1)
+            X_test_3d = X_test.reshape(X_test.shape[0], X_test.shape[1], 1)
+        except IndexError:
+            return LSTMData(
+                code=self.code,
+                data_2d=np.array([]),
+                train_size=0,
+                train_data_2d=np.array([]),
+                test_data_2d=np.array([]),
+                X_train_3d=np.array([]),
+                X_test_3d=np.array([]),
+                y_train_1d=np.array([]),
+                y_test_1d=np.array([]),
+            )
 
         tsa_logger.debug(f'n_dim - X_train_3d : {X_train_3d.ndim}, X_test_3d : {X_test_3d.ndim}, y_train : {y_train_1d.ndim}, y_test : {y_test_1d.ndim}')
         tsa_logger.debug(f'len - X_train_3d : {len(X_train_3d)}, X_test_3d : {len(X_test_3d)}, y_train : {len(y_train_1d)}, y_test : {len(y_test_1d)}')
@@ -517,8 +540,11 @@ class MyLSTM:
             :return:
             """
             # 앙상블 테스트와 채점
-            _, _, ensemble_future_predictions_2d = self.ensemble_training(
+            try:
+                _, _, ensemble_future_predictions_2d = self.ensemble_training(
                 num=num_in)
+            except IndexError:
+                return [], []
 
             """if grading:
                 lstm_grade = self.grading(ensemble_train_predictions_2d, ensemble_test_predictions_2d)

{analyser_hj3415-2.9.18.dist-info → analyser_hj3415-2.10.1.dist-info}/METADATA

@@ -1,11 +1,11 @@
 Metadata-Version: 2.3
 Name: analyser_hj3415
-Version: 2.9.18
+Version: 2.10.1
 Summary: Stock analyser and database processing programs
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
-Requires-Dist: utils-hj3415>=2.9.2
-Requires-Dist: db-hj3415>=4.1.4
+Requires-Dist: utils-hj3415>=2.9.4
+Requires-Dist: db-hj3415>=4.2.3
 Requires-Dist: scikit-learn>=1.5.2
 Requires-Dist: plotly>=5.24.1
 Requires-Dist: yfinance>=0.2.44

{analyser_hj3415-2.9.18.dist-info → analyser_hj3415-2.10.1.dist-info}/RECORD

@@ -1,13 +1,13 @@
 analyser_hj3415/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 analyser_hj3415/cli.py,sha256=a4oUwfPFEdd3r2Fq2EMHBITcp6lbXUjZvVWvCHIvCUQ,12540
 analyser_hj3415/eval.py,sha256=eNrcbpVyj7SZJevWqatlj_gJ2EQTmLZAz2nPG5qNv6k,38811
-analyser_hj3415/tsa.py,sha256=A4iFhNGXAuxEvW-Jn6lpMY7JJyKil4Q0XYA31-dUOXg,27882
+analyser_hj3415/tsa.py,sha256=gbpS4ZXgHYlN1MggqA3UUiTkmz8-WvYkxto_yjQt13g,28558
 analyser_hj3415/workroom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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.9.18.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
-analyser_hj3415-2.9.18.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-analyser_hj3415-2.9.18.dist-info/METADATA,sha256=ZVd22jJkK5iLGAv_SbiiXI3OefzIy69M4ifg2YtgQVQ,6777
-analyser_hj3415-2.9.18.dist-info/RECORD,,
+analyser_hj3415-2.10.1.dist-info/entry_points.txt,sha256=ZfjPnJuH8SzvhE9vftIPMBIofsc65IAWYOhqOC_L5ck,65
+analyser_hj3415-2.10.1.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+analyser_hj3415-2.10.1.dist-info/METADATA,sha256=HDWRHN49PtQQ0rzT2L9QObehx9r1E773C0vj2hQfcIc,6777
+analyser_hj3415-2.10.1.dist-info/RECORD,,