Skip to content

Results

Data structures for results

TrainResult

Bases: BaseModel

Source code in src/fed_rag/data_structures/results.py 
8
9
class TrainResult(BaseModel):
    loss: float

TestResult

Bases: BaseModel

Source code in src/fed_rag/data_structures/results.py 
12
13
14
15
16
17
18
19
20
class TestResult(BaseModel):
    __test__ = (
        False  # needed for Pytest collision. Avoids PytestCollectionWarning
    )
    loss: float
    metrics: dict[str, Any] = Field(
        description="Additional metrics computed on test set.",
        default_factory=dict,
    )