罗俊辉 před 1 rokem
rodič
revize
62c7aabaaa
1 změnil soubory, kde provedl 2 přidání a 1 odebrání
  1. 2 1
      main.py

+ 2 - 1
main.py

@@ -147,7 +147,7 @@ class LightGBM(object):
         """
         fw = open("result/summary_{}.txt".format(dt), "a+", encoding="utf-8")
         path = 'data/predict_data/all_predict_{}.json'.format(dt)
-        x, y, ori_df = self.read_data(path, yc=6)
+        x, y, ori_df = self.read_data(path, yc=0.02)
         true_label_df = pd.DataFrame(list(y), columns=['ture_label'])
         bst = lgb.Booster(model_file=self.model)
         y_pred = bst.predict(x, num_iteration=bst.best_iteration)
@@ -194,6 +194,7 @@ class LightGBM(object):
 # "cat summary_20240326.txt | awk -F "\t" '{print $1" "$3}'| /root/AUC/AUC/AUC"
 """
  ossutil64 cp /root/luojunhui/alg/data/predict_data/spider_predict_result_20240330.xlsx oss://art-pubbucket/0temp/
+ cat summary_20240327.txt | awk -F "\t" '{print $1" "$3}'| /root/AUC/AUC/AUC
 """
 if __name__ == "__main__":
     i = int(input("输入 1 训练, 输入 2 预测:\n"))