|
|
@@ -4,6 +4,7 @@ import logging
|
|
|
from datetime import datetime, timedelta
|
|
|
|
|
|
import pandas as pd
|
|
|
+import numpy as np
|
|
|
import xgboost as xgb
|
|
|
|
|
|
from client import ODPSClient
|
|
|
@@ -330,6 +331,14 @@ def _main():
|
|
|
|
|
|
filtered_vid = predict_df.loc[condition_choose, 'vid'].unique()
|
|
|
|
|
|
+ # 写入本地文件
|
|
|
+ np.savetxt(
|
|
|
+ f"{config_manager.project_home}/XGB/file/filtered_vid_{datetime.now().strftime('%Y%m%d')}.csv",
|
|
|
+ filtered_vid,
|
|
|
+ fmt="%d",
|
|
|
+ delimiter=","
|
|
|
+ )
|
|
|
+
|
|
|
# 写入Redis
|
|
|
redis_key = f"redis:lower_vov_vid:{datetime.now().strftime('%Y%m%d')}"
|
|
|
|
