Ver código fonte

Change model remport generator fiel

Mohidul Islam 4 anos atrás
pai
commit
d6a61fb802
1 arquivos alterados com 2 adições e 0 exclusões
  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]