VJ预测局部调整

data_preprocess.py

@@ -1200,14 +1200,14 @@ def predict_data_simple(df_input, group_route_str, output_dir, predict_dir=".",
     ).round(4)
 
     # 综合评分阈值：大于阈值的都认为值得投放
-    target_score_threshold = 0.8
+    target_score_threshold = 0.7
     # df_min_hours['target_score_threshold'] = target_score_threshold
     df_min_hours['is_good_target'] = (df_min_hours['target_score'] >= target_score_threshold).astype(int)
 
     print(f">>> 包络线+降价潜力评分完成")
     del df_hist, df_future, df_envelope_all, df_envelope_agg, df_top, df_drop_freq, df_rise_freq
     
-    df_min_hours = df_min_hours[df_min_hours['is_good_target'] == 1].reset_index(drop=True)   # 保留值得投放的 
+    df_min_hours = df_min_hours[(df_min_hours['is_good_target'] == 1) & (df_min_hours['seats_remaining'] >= 5)].reset_index(drop=True)   # 保留值得投放的 
 
     # =====================================================================
 

descending_cabin_task.py

@@ -653,13 +653,17 @@ def main():
     # logger.info(f"policy_list: {policy_list}")
     logger.info(f"policy_list: {json.dumps(policy_list, ensure_ascii=False, default=str)[:1000]}")
     if len(policy_list) > 0:
-        # 这里批量一次性上传政策 
+        # 这里批量一次性上传政策
         payload = {"items": policy_list}
-        sync_policy(payload)
+        try:
+            sync_policy(payload)
+            logger.info(f"上传政策成功")
+        except Exception as e:
+            logger.error(f"上传政策失败: {e}")
 
     logger.info(f"keep_info_end: {len(keep_info_end)}")
     # 将 keep_info_end 写入到文件csv 文件 嵌套结构要处理  提供下载页面 (历史数据需要保留)
-    output_dir = "/home/node04/descending_cabin_files"
+    output_dir = "/home/node04/descending_cabin_files_vj"
     os.makedirs(output_dir, exist_ok=True)
     if keep_info_end:
         out_path = os.path.join(

main_tr_0.py

@@ -50,7 +50,7 @@ def start_train():
     # date_end = datetime.today().strftime("%Y-%m-%d")
     date_end = (datetime.today() - timedelta(days=1)).strftime("%Y-%m-%d")
     # date_begin = (datetime.today() - timedelta(days=32)).strftime("%Y-%m-%d")
-    date_begin = "2026-01-01"   # 2026-01-01 2026-03-11 2026-03-16 2026-03-18
+    date_begin = "2026-04-05"   # 2026-01-01 2026-03-18 2026-04-05 2026-04-07
 
     print(f"训练时间范围: {date_begin} 到 {date_end}")
 

result_keep_verify.py

@@ -377,4 +377,4 @@ def verify_process_2(min_batch_time_str, max_batch_time_str):
 
 if __name__ == "__main__":
     # verify_process("202604021500", "202604030900")
-    verify_process_2("202604021700", "202604031600")
+    verify_process_2("202604031700", "202604071300")