您好,文本分类推理的写法能否优化。
Created by: chenfengshf
鉴于目前,那边的平台只能一条一条的载入文本预测,,,所以推理速度慢的想哭。 我觉得是我的写法存在漏洞。 想问问有没有什么优化措施,请不吝赐教。
def predict(self, title, text):
'''
模型预测返回结果
:param input: 评估传入样例 {"title": "心率为72bpm是正常的吗", "text": "最近不知道怎么回事总是感觉心脏不舒服..."}
:return: 模型预测成功中户 {"label": "心血管科"}
'''
text_line = title+text
predict_data = [[ text_line ]]
run_states = self.cls_task.predict(data=predict_data)
results = [run_state.run_results for run_state in run_states]
pred = self.label_list[np.argmax(results[0][0][0])]
return {'label': pred}