fix issue with number of classes in EMNIST

enriquetomasmb · enriquetomasmb · commit 9f57d46845ab · 2025-05-07T16:51:47.000+02:00
diff --git a/nebula/core/models/emnist/cnn.py b/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,
diff --git a/nebula/core/models/emnist/mlp.py b/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,
diff --git a/nebula/scenarios.py b/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,

Original file line number	Diff line number	Diff line change
`@@ -639,7 +639,7 @@ def load_configurations_and_start_nodes(self, additional_participants=None, sche`
`639`	`639`	`)`
`640`	`640`	`elif dataset_name == "EMNIST":`
`641`	`641`	`dataset = EMNISTDataset(`
`642`		`- num_classes=10,`
	`642`	`+ num_classes=47,`
`643`	`643`	`partitions_number=self.n_nodes,`
`644`	`644`	`iid=self.scenario.iid,`
`645`	`645`	`partition=self.scenario.partition_selection,`