فهرست منبع

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

myhloli 3 ماه پیش
والد
کامیت
2a8e6f9d45
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      mineru/model/table/cls/paddle_table_cls.py

+ 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]