diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 8943cecd7009608b10ced44146ad140b2ffcb2c2..271f77f05a0686b6124d70bd321aa15a6137a888 100644 --- a/pysegcnn/core/trainer.py +++ b/pysegcnn/core/trainer.py @@ -2652,6 +2652,7 @@ class NetworkInference(BaseConfig): in output.items()]).flatten() # calculate confusion matrix + LOGGER.info('Computing confusion matrix ...') conf_mat = confusion_matrix(y_true, y_pred) # add confusion matrix to model output