Spaces:
Sleeping
Sleeping
Commit ·
c6d89e5
1
Parent(s): 18e31c4
Upload infer.py
Browse files
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
|