diff --git a/utils/lgb_predict_task2.py b/utils/lgb_predict_task2.py index 3647a1267f8f744cbb635ca22bcf0ae29d8edfc7..c899e3926074ce048b95de232f8bb75cdff3d80f 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']