diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index 7ed0bb57a847f984f37d3da9c05a0f246ab5d2ec..ba9df9635831fec44bd81ce803c6d92147299a20 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -634,7 +634,7 @@ 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.float32)
+        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'}
 
         # iterate over the samples of the dataset