Fix: Bug on prediction for CRF model (#16)

monologg · Aug 29, 2020 · a822e9f · a822e9f
1 parent 8bfa415
commit a822e9f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/predict.py b/predict.py
@@ -184,9 +184,7 @@ def predict(pred_config):
 
     intent_preds = np.argmax(intent_preds, axis=1)
 
-    if args.use_crf:
-        slot_preds = np.array(model.crf.decode(slot_logits))
-    else:
+    if not args.use_crf:
         slot_preds = np.argmax(slot_preds, axis=2)
 
     slot_label_map = {i: label for i, label in enumerate(slot_label_lst)}