From dc8b0ca130f4839cceac42d6de56fec8a9dd4c60 Mon Sep 17 00:00:00 2001
From: wufanyou <fanyou.wu@outlook.com>
Date: Mon, 11 Jul 2022 11:55:12 +0800
Subject: [PATCH] update
---
utils/lgb_predict_task2.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/utils/lgb_predict_task2.py b/utils/lgb_predict_task2.py
index 3647a12..c899e39 100755
--- a/utils/lgb_predict_task2.py
+++ b/utils/lgb_predict_task2.py
@@ -7,21 +7,21 @@ __MAP__ = ["irrelevant", "complement", "substitute", "exact"]
LGB_CONFIG = {
"us": {
"product_feat": "/models/us-product-feat.csv",
- "model_file": "/models/lgb-us.txt",
+ "model_file": "/models/lgb-us-task-2.txt",
"features": ['pred_0','pred_1','pred_2','pred_3',
'label_0','label_1','label_2','label_3',
'query_count','is_isbn','has_isbn','fold']
},
"jp": {
"product_feat": "/models/jp-product-feat.csv",
- "model_file": "/models/lgb-es-jp.txt",
+ "model_file": "/models/lgb-es-jp-task-2.txt",
"features": ['pred_0','pred_1','pred_2','pred_3',
'label_0','label_1','label_2','label_3',
'query_count','is_isbn','has_isbn','fold','locale']
},
"es": {
"product_feat": "/models/es-product-feat.csv",
- "model_file": "/models/lgb-es-jp.txt",
+ "model_file": "/models/lgb-es-jp-task-2.txt",
"features": ['pred_0','pred_1','pred_2','pred_3',
'label_0','label_1','label_2','label_3',
'query_count','is_isbn','has_isbn','fold','locale']
--
GitLab