Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

提高文本分类精度

Browse Source
This commit is contained in:
liuyebo
2024-10-18 11:02:58 +08:00
parent 3f93bd476a
commit 6529dc3d98
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
services/paddle_services/clas_text.py
View File

@@ -9,7 +9,7 @@ from utils import process_request
app = Flask(__name__) app = Flask(__name__)
schema = ['基本医保结算单', '出院记录', '费用清单'] schema = ['基本医保结算单', '出院记录', '费用清单']
CLAS = Taskflow('zero_shot_text_classification', model='utc-xbase', schema=schema, CLAS = Taskflow('zero_shot_text_classification', model='utc-xbase', schema=schema,
task_path='model/text_classification', precision='fp16') task_path='model/text_classification', precision='fp32')
@app.route('/', methods=['POST']) @app.route('/', methods=['POST'])
@@ -20,7 +20,7 @@ def main():
cls_result = cls_result[0].get('predictions') cls_result = cls_result[0].get('predictions')
if cls_result: if cls_result:
cls_result = cls_result[0] cls_result = cls_result[0]
if cls_result['score'] < 0.8: if cls_result['score'] and float(cls_result['score']) < 0.8:
logging.info(f"识别结果置信度{cls_result['score']}过低！text: {text}") logging.info(f"识别结果置信度{cls_result['score']}过低！text: {text}")
return None return None
return cls_result['label'] return cls_result['label']