Upload infer.py

infer.py

@@ -110,9 +110,9 @@ def infer_3_class(model, tokenizer, index, label_dict, is_mixed_dict, text, K):
         # pred.append(final)
         total_score = sum(fuzzy_cnt.values())
         final = dict()
-        final[0] = int(fuzzy_cnt[(1,0)] / total_score*100)
-        final[1] = int(fuzzy_cnt[(0,10^3)] / total_score*100)
-        final[2] = int(fuzzy_cnt[(1,1)] / total_score*100)
+        final[0] = round(fuzzy_cnt[(1,0)] / total_score*100,2)
+        final[1] = round(fuzzy_cnt[(0,10^3)] / total_score*100,2)
+        final[2] = round(fuzzy_cnt[(1,1)] / total_score*100,2)
         print(f"Final prediction: {final}")
     return final