diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 757fb3f37e2492be8db4b394997d1ba717232ee1..cbe4d0fadd2fef89537832bb7e9eb443f1c62df4 100644 --- a/pysegcnn/core/trainer.py +++ b/pysegcnn/core/trainer.py @@ -435,6 +435,8 @@ class EvalConfig(BaseConfig): plot_bands: list = dataclasses.field( default_factory=lambda: ['nir', 'red', 'green']) cm: bool = True + figsize: tuple = (10, 10) + alpha: int = 5 def __post_init__(self): super().__post_init__() @@ -477,6 +479,7 @@ class NetworkTrainer(BaseConfig): self.es = EarlyStopping(self.mode, self.max_accuracy, self.delta, self.patience) + def train(self): LOGGER.info(30 * '-' + ' Training ' + 30 * '-')