Commit 487789ef by 王金柱

修改咨询师返回值数据类型为 string 类型

parent 2a2687cd
...@@ -302,7 +302,7 @@ class RecommendByXgboost(Recommender): ...@@ -302,7 +302,7 @@ class RecommendByXgboost(Recommender):
result_dict = dict(zip(doctor_ids, predit_result)) result_dict = dict(zip(doctor_ids, predit_result))
result_dict = sorted(result_dict.items(), key=lambda x:x[1], reverse=True) result_dict = sorted(result_dict.items(), key=lambda x:x[1], reverse=True)
recommend_data = [{ recommend_data = [{
'counselor': int(c_id), 'counselor': c_id,
'score': round(float(proba), 4), 'score': round(float(proba), 4),
'from': 'similar_users {}'.format(user_id), 'from': 'similar_users {}'.format(user_id),
} for (c_id, proba) in result_dict[0:50]] } for (c_id, proba) in result_dict[0:50]]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment