瀏覽代碼

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]