Quellcode durchsuchen

Change model remport generator fiel

Mohidul Islam vor 4 Jahren
Ursprung
Commit
d6a61fb802
1 geänderte Dateien mit 2 neuen und 0 gelöschten Zeilen
  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]