|
@@ -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"))
|