From 1ddf595bbaf79accfabc3eaa1b332d8ae43b505d Mon Sep 17 00:00:00 2001
From: wufanyou <fanyou.wu@outlook.com>
Date: Sun, 10 Jul 2022 19:30:59 +0800
Subject: [PATCH] update
---
utils/lgb_predict.py | 2 +-
utils/run.py | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/utils/lgb_predict.py b/utils/lgb_predict.py
index 1546b02..010cab8 100755
--- a/utils/lgb_predict.py
+++ b/utils/lgb_predict.py
@@ -32,7 +32,7 @@ LGB_CONFIG = {
MAP = ["irrelevant", "complement", "substitute", "exact"]
LOCALE_MAP = {'jp':0, 'es':1, 'us':2}
COL_NAME = "esci_label"
-WEIGHT_MAP = {0:0.4, 1:0.4, 2:0.4, 3:0.4, 4:0.2, 5:0.2}
+WEIGHT_MAP = {0:0.4, 1:0.4, 2:0.4, 3:0.4, 4:0.25, 5:0.2}
def lgb_predict(df, locale):
df = df.reset_index(drop=True)
diff --git a/utils/run.py b/utils/run.py
index 94a5ab6..2bac491 100755
--- a/utils/run.py
+++ b/utils/run.py
@@ -57,7 +57,8 @@ CONFIG = {
"product_brand": 42452,
"product_color_name": 44287,
},
- "model": ["/models/us-distilbart-two-fold-0-fp16.onnx" ,"/models/us-distilbart-two-fold-1-fp16.onnx"],
+ "model": ["/models/us-distilbart-two-fold-0-fp16.onnx"],
+ #"model": ["/models/us-distilbart-two-fold-0-fp16.onnx" ,"/models/us-distilbart-two-fold-1-fp16.onnx"],
"product": "/models/distilbart.h5",
"key": "us",
"type": "distilbart",
--
GitLab