diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index ba9df9635831fec44bd81ce803c6d92147299a20..b4ed855e27773db85038c17e33413c655d1e42ee 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -634,8 +634,8 @@ class ImageDataset(Dataset):
 
         # initialize dictionary of class spectral distribution
         # exclude NoData class
-        cls_ds = {k: np.empty(shape=(0, len(self.use_bands)), dtype=np.float16)
-                  for k, v in self.labels.items() if v['label'] != 'No_data'}
+        cls_ds = {k: np.empty(shape=(0, len(self.use_bands)), dtype=np.float32)
+                  for k, v in self.labels.items() if v['label'] != 'No data'}
 
         # iterate over the samples of the dataset
         for i in range(len(self)):