fix issue with number of classes in EMNIST

Author: enriquetomasmb

nebula/core/models/emnist/cnn.py

@@ -7,7 +7,7 @@ class EMNISTModelCNN(NebulaModel):
     def __init__(
         self,
         input_channels=1,
-        num_classes=10,
+        num_classes=47,
         learning_rate=1e-3,
         metrics=None,
         confusion_matrix=None,

nebula/core/models/emnist/mlp.py

@@ -7,7 +7,7 @@ class EMNISTModelMLP(NebulaModel):
     def __init__(
         self,
         input_channels=1,
-        num_classes=10,
+        num_classes=47,
         learning_rate=1e-3,
         metrics=None,
         confusion_matrix=None,

nebula/scenarios.py

@@ -639,7 +639,7 @@ def load_configurations_and_start_nodes(self, additional_participants=None, sche
             )
         elif dataset_name == "EMNIST":
             dataset = EMNISTDataset(
-                num_classes=10,
+                num_classes=47,
                 partitions_number=self.n_nodes,
                 iid=self.scenario.iid,
                 partition=self.scenario.partition_selection,