davesalvi
/

ispl_safe

Model card Files Files and versions

davesalvi commited on Apr 3, 2025

Commit

4070d13

·

1 Parent(s): 6bb7746

change thres

Files changed (1) hide show

src/lcnn_model.py +1 -1

src/lcnn_model.py CHANGED Viewed

@@ -24,7 +24,7 @@ class LCNN(nn.Module):
     def __init__(self, return_emb=False, num_class=2):
         super(LCNN, self).__init__()
-        self.threshold = 0.15
         self.return_emb = return_emb
         # Feature Extraction Part (First Part)

     def __init__(self, return_emb=False, num_class=2):
         super(LCNN, self).__init__()
+        self.threshold = 0.25
         self.return_emb = return_emb
         # Feature Extraction Part (First Part)