Преглед на файлове

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]