Merge pull request #32 from Enraged-Dun-Cookie-Development-Team/auto-…

…scheduler-v0.1 增大时间间隔；先预测2个小时的数据点
Enraged-Dun-Cookie-Development-Team · Oct 19, 2024 · aed18d8 · aed18d8
2 parents f1b6717 + d43a41a
commit aed18d8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/conf/auto_sche.conf b/conf/auto_sche.conf
@@ -1,6 +1,6 @@
 {
     "daily_preprocess_time": {
-        "hour": 20,
+        "hour": 21,
         "minute": 0,
         "second": 0
     },

diff --git a/src/_data_lib.py b/src/_data_lib.py
@@ -336,7 +336,7 @@ def limit_cpu(interval):
             messager.send_to_bot_shortcut('开始预测')
 
             for i in range(0, len(X_list), batch_size):
-                time.sleep(0.02)
+                time.sleep(0.05)
                 batch = X_list[i:i + batch_size]
                 batch_predictions = self.model.predict(batch)
                 predictions.extend(batch_predictions)

diff --git a/src/auto_sche/model_events.py b/src/auto_sche/model_events.py
@@ -87,7 +87,7 @@ def feature_of_time(self):
         #     scheduled_time += datetime.timedelta(days=1)
 
         # 生成未来24小时的时间点，从凌晨4点开始，到凌晨4点结束（不包括）
-        end_time = scheduled_time + datetime.timedelta(days=1)
+        end_time = scheduled_time + datetime.timedelta(hours=2)
         time_points = self._generate_time_points(scheduled_time, end_time)
 
         return time_points