Эх сурвалжийг харах

fix: increase confidence threshold for table classification in paddle_table_cls.py

myhloli 3 сар өмнө
parent
commit
2a8e6f9d45

+ 1 - 1
mineru/model/table/cls/paddle_table_cls.py

@@ -68,6 +68,6 @@ class PaddleTableClsModel:
         idx = np.argmax(result)
         conf = float(np.max(result))
         # logger.debug(f"Table classification result: {self.labels[idx]} with confidence {conf:.4f}")
-        if idx == 0 and conf < 0.8:
+        if idx == 0 and conf < 0.9:
             idx = 1
         return self.labels[idx]