From 4460601c748e2024ebcaca6c8ea38eb041f9bed9 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 16 Feb 2021 14:35:51 +0100
Subject: [PATCH] Initialize DataFrame with numeric type.
---
pysegcnn/core/trainer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index cfdc4b4..2d97f16 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2730,7 +2730,7 @@ class NetworkInference(BaseConfig):
['{}'.format(mstate.name) for mstate in self.state_files]))
# iterate over the individual models
- df = pd.DataFrame()
+ df = pd.DataFrame(dtype=float)
for name, output in inference.items():
# classification report DataFrame for all individual models
df = pd.concat((df, df))
--
GitLab