|
@@ -30,7 +30,7 @@ azmodel = FlagModel('/mnt/d/Develop/az/az_encoder_only_base_bge-large-zh-v1.5')
|
|
|
model = None # FlagModel('/mnt/d/Develop/az/az_encoder_only_base_bge-large-zh-v1.5')
|
|
model = None # FlagModel('/mnt/d/Develop/az/az_encoder_only_base_bge-large-zh-v1.5')
|
|
|
cuoshi_model = None #FlagModel('cuoshi_encoder_only_base_bge-large-zh-v1.5/cuoshi_encoder_only_base_bge-large-zh-v1.5')
|
|
cuoshi_model = None #FlagModel('cuoshi_encoder_only_base_bge-large-zh-v1.5/cuoshi_encoder_only_base_bge-large-zh-v1.5')
|
|
|
from sentence_transformers import CrossEncoder
|
|
from sentence_transformers import CrossEncoder
|
|
|
-azce = CrossEncoder('/mnt/d/Develop/az/models/reranker-chinese-macbert-large-gooaq-bce/final')
|
|
|
|
|
|
|
+azce = CrossEncoder('/mnt/d/Develop/az/models/reranker-chinese-macbert-large-gooaq-bce-3/final')
|
|
|
ce = None #CrossEncoder('/mnt/d/Develop/az/models/reranker-chinese-macbert-large-gooaq-bce/final')
|
|
ce = None #CrossEncoder('/mnt/d/Develop/az/models/reranker-chinese-macbert-large-gooaq-bce/final')
|
|
|
cuoshi_ce = None #CrossEncoder('cuoshi_reranker/final')
|
|
cuoshi_ce = None #CrossEncoder('cuoshi_reranker/final')
|
|
|
with open("hunningtu_rule", "r") as f:
|
|
with open("hunningtu_rule", "r") as f:
|
|
@@ -398,8 +398,8 @@ def transform(answer, input, entry):
|
|
|
answer2 = answer2.replace("\"", "")
|
|
answer2 = answer2.replace("\"", "")
|
|
|
return answer2
|
|
return answer2
|
|
|
|
|
|
|
|
-def zuhe(input, content):
|
|
|
|
|
- if input in incremental and input != '17-176' and input != '17-177' and input != '17-179' and input != '17-180' and input != '17-181' and input != '15-33' and input != '1534' and input != '15-35' and input != '15-36' and input != '15-37' and input != '15-38' and input != '15-39' and input != '15-40' and input != '15-41':
|
|
|
|
|
|
|
+def zuhe(input, content, bianma):
|
|
|
|
|
+ if bianma.startswith('01') and input in incremental and input != '17-176' and input != '17-177' and input != '17-179' and input != '17-180' and input != '17-181' and input != '15-33' and input != '1534' and input != '15-35' and input != '15-36' and input != '15-37' and input != '15-38' and input != '15-39' and input != '15-40' and input != '15-41':
|
|
|
option = incremental[input]
|
|
option = incremental[input]
|
|
|
for entry in option:
|
|
for entry in option:
|
|
|
answer = callzai(name_label[input], name_label[entry], content)
|
|
answer = callzai(name_label[input], name_label[entry], content)
|
|
@@ -488,6 +488,13 @@ def huansuan(input, dw, label):
|
|
|
return 0.1
|
|
return 0.1
|
|
|
if dw1 == 'kg' and dw2 == '100kg':
|
|
if dw1 == 'kg' and dw2 == '100kg':
|
|
|
return 0.01
|
|
return 0.01
|
|
|
|
|
+ if dw1 == '个' and dw2 == '套':
|
|
|
|
|
+ return 1
|
|
|
|
|
+ if dw1 == '个' and dw2 == '10套':
|
|
|
|
|
+ return 0.1
|
|
|
|
|
+ if dw1 == 'm' and dw2 == '100m单线':
|
|
|
|
|
+ return 0.01
|
|
|
|
|
+
|
|
|
else:
|
|
else:
|
|
|
answer = callzaidw(dw1,dw2)
|
|
answer = callzaidw(dw1,dw2)
|
|
|
answers = answer.split("\n")
|
|
answers = answer.split("\n")
|
|
@@ -510,8 +517,10 @@ def huansuan(input, dw, label):
|
|
|
answer2=answer2.replace("*", "")
|
|
answer2=answer2.replace("*", "")
|
|
|
answer2=answer2.replace("x", "")
|
|
answer2=answer2.replace("x", "")
|
|
|
answer2=answer2.replace("×", "")
|
|
answer2=answer2.replace("×", "")
|
|
|
- print(answer2)
|
|
|
|
|
- return float(answer2)
|
|
|
|
|
|
|
+ try:
|
|
|
|
|
+ return float(answer2)
|
|
|
|
|
+ except:
|
|
|
|
|
+ return 1
|
|
|
|
|
|
|
|
def clarify(data):
|
|
def clarify(data):
|
|
|
data['tz'] = data['tz'].replace('DTG', '')
|
|
data['tz'] = data['tz'].replace('DTG', '')
|
|
@@ -842,7 +851,7 @@ def process_data(data:dict)-> dict:
|
|
|
result = [(label_name[x], huansuan_highlevel(data['bianma'], x, data['dw'], label_name[x], data['mc']+data['tz'])) for x in selected]
|
|
result = [(label_name[x], huansuan_highlevel(data['bianma'], x, data['dw'], label_name[x], data['mc']+data['tz'])) for x in selected]
|
|
|
print("after haunsuan")
|
|
print("after haunsuan")
|
|
|
print(result)
|
|
print(result)
|
|
|
- result = [(zuhe(x[0], label).replace(',',''), x[1]) for x in result]
|
|
|
|
|
|
|
+ result = [(zuhe(x[0], label, data['bianma']).replace(',',''), x[1]) for x in result]
|
|
|
print("after zuhe")
|
|
print("after zuhe")
|
|
|
print(result)
|
|
print(result)
|
|
|
if data['bianma'].startswith('01'):
|
|
if data['bianma'].startswith('01'):
|