From 6c2a33fe51e4844f2e4eed416eceb6a43d5272f4 Mon Sep 17 00:00:00 2001
From: cuicheng01 <45199522+cuicheng01@users.noreply.github.com>
Date: Sun, 25 Apr 2021 18:48:26 +0800
Subject: [PATCH] fix warnings bug (#698)

---
 ppcls/utils/metrics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppcls/utils/metrics.py b/ppcls/utils/metrics.py
index 724cae2d..b0db68a7 100644
--- a/ppcls/utils/metrics.py
+++ b/ppcls/utils/metrics.py
@@ -33,7 +33,7 @@ def multi_hot_encode(logits, threshold=0.5):
     Encode logits to multi-hot by elementwise for multilabel
     """
 
-    return binarize(logits, threshold)
+    return binarize(logits, threshold=threshold)
 
 
 def hamming_distance(output, target):
-- 
GitLab