From cd548a440b79739cc8e4426545f171a4ee5739fd Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Mon, 17 Aug 2020 10:21:18 +0200
Subject: [PATCH] Adjusted EvalConfig
---
pysegcnn/core/trainer.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 757fb3f..cbe4d0f 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 * '-')
--
GitLab