ソースを参照

Change model remport generator fiel

Mohidul Islam 4 年 前
コミット
d6a61fb802
1 ファイル変更2 行追加0 行削除
  1. 2 0
      review/model_pred_report.py

+ 2 - 0
review/model_pred_report.py

@@ -59,6 +59,8 @@ def get_review_report_of_nth_days(days):
             review_text = review.comment
             reply = review.reply.replied_text
             model_pred = model_inference(review_text)
+            if model_pred:
+                model_pred = model_pred[0]
             actual_class = get_review_actual_intent(reply)
             classified = do_predict_correctly(review, actual_class)
             row = [review_text, reply, model_pred, actual_class, classified]