ngocminhta commited on
Commit
c6d89e5
·
1 Parent(s): 18e31c4

Upload infer.py

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -111,7 +111,7 @@ def infer_3_class(model, tokenizer, index, label_dict, is_mixed_dict, text, K):
111
  total_score = sum(fuzzy_cnt.values())
112
  final = dict()
113
  final[0] = int(fuzzy_cnt[(1,0)] / total_score*100)
114
- final[1] = int(fuzzy_cnt[(0,10^3)] / total_score*100
115
  final[2] = int(fuzzy_cnt[(1,1)] / total_score*100)
116
  print(f"Final prediction: {final}")
117
  return final
 
111
  total_score = sum(fuzzy_cnt.values())
112
  final = dict()
113
  final[0] = int(fuzzy_cnt[(1,0)] / total_score*100)
114
+ final[1] = int(fuzzy_cnt[(0,10^3)] / total_score*100)
115
  final[2] = int(fuzzy_cnt[(1,1)] / total_score*100)
116
  print(f"Final prediction: {final}")
117
  return final