mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 04:24:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move grade to page

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-05-06 15:33:25 +02:00
parent a8a8b8e0f9
commit f9496e4a91
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
docling/datamodel/base_models.py
View File

@ -306,6 +306,14 @@ class OpenAiApiResponse(BaseModel):
ScoreValue = float ScoreValue = float
class QualityGrade(str, Enum):
POOR = "poor"
FAIR = "fair"
GOOD = "good"
EXCELLENT = "excellent"
UNSPECIFIED = "unspecified"
class PageConfidenceScores(BaseModel): class PageConfidenceScores(BaseModel):
overall_score: ScoreValue = np.nan overall_score: ScoreValue = np.nan
@ -314,16 +322,6 @@ class PageConfidenceScores(BaseModel):
table_score: ScoreValue = np.nan table_score: ScoreValue = np.nan
ocr_score: ScoreValue = np.nan ocr_score: ScoreValue = np.nan
class QualityGrade(str, Enum):
POOR = "poor"
FAIR = "fair"
GOOD = "good"
VERY_GOOD = "very_good"
UNSPECIFIED = "unspecified"
class ConfidenceReport(PageConfidenceScores):
@computed_field # type: ignore @computed_field # type: ignore
@property @property
def grade(self) -> QualityGrade: def grade(self) -> QualityGrade:
@ -334,10 +332,12 @@ class ConfidenceReport(PageConfidenceScores):
elif self.overall_score < 0.9: elif self.overall_score < 0.9:
return QualityGrade.GOOD return QualityGrade.GOOD
elif self.overall_score >= 0.9: elif self.overall_score >= 0.9:
return QualityGrade.VERY_GOOD return QualityGrade.EXCELLENT
return QualityGrade.UNSPECIFIED return QualityGrade.UNSPECIFIED
class ConfidenceReport(PageConfidenceScores):
pages: Dict[int, PageConfidenceScores] = Field( pages: Dict[int, PageConfidenceScores] = Field(
default_factory=lambda: defaultdict(PageConfidenceScores) default_factory=lambda: defaultdict(PageConfidenceScores)
) )