from .model_wrappers import BaseLightningWrapper
[docs]
class CorrectionResult:
""" """
def __init__(self, method: str, model: BaseLightningWrapper, metrics: dict) -> None:
self.method = method
self.model = model
self.metrics = metrics
def __str__(self):
return f"Results for: {self.method}"
def __repr__(self):
return self.__str__()
[docs]
def get_all_metrics(self) -> dict:
""" """
return self.metrics
[docs]
def get_metric(self, metric: str) -> float:
"""
Args:
metric: str:
Returns:
"""
return self.metrics["all"][metric]
[docs]
def get_model(self) -> BaseLightningWrapper:
""" """
return self.model
[docs]
def get_method(self) -> str:
""" """
return self.method