diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 2d97f16d7fdf38de2f2f8d1e32260fc389c36c98..49349d8608be7f73e5fe6841a0c4b0cb15bb5b5a 100644 --- a/pysegcnn/core/trainer.py +++ b/pysegcnn/core/trainer.py @@ -2733,7 +2733,7 @@ class NetworkInference(BaseConfig): df = pd.DataFrame(dtype=float) for name, output in inference.items(): # classification report DataFrame for all individual models - df = pd.concat((df, df)) + df = pd.concat([df, output['report']], axis=0) # compute k-fold average estimate of each metric across all models report = df.groupby(df.index, sort=False).mean()