Add feature shape and class count methods to HiggsDataLoader (#1641)

This update introduces two new methods, `get_feature_shape` and `get_num_classes`, to the HiggsDataLoader class. These methods provide easy access to the feature shape and the number of classes for the Higgs dataset, enhancing the usability of the data loader.

Signed-off-by: Rahul Garg <rahul.garg@intel.com>

Author: rahulga1

openfl-workspace/xgb_higgs/src/dataloader.py

@@ -25,6 +25,9 @@ class HiggsDataLoader(XGBoostDataLoader):
     def __init__(self, data_path=None, **kwargs):
         super().__init__(**kwargs)
 
+        # Define default feature shape and number of classes for Higgs dataset
+        self.feature_shape = (28,)
+        self.num_classes = 2
 
         # If data_path is None, this is being used for model initialization only
         if data_path is None:
@@ -38,6 +41,22 @@ def __init__(self, data_path=None, **kwargs):
         self.X_valid = X_valid
         self.y_valid = y_valid
 
+    def get_feature_shape(self):
+        """Returns the shape of an example feature array.
+
+        Returns:
+            list: The shape of an example feature array [3, 150, 150] for Histology images.
+        """
+        return self.feature_shape
+
+    def get_num_classes(self):
+        """Returns the number of classes for classification tasks.
+
+        Returns:
+            int: The number of classes (8 for Histology dataset).
+        """
+        return self.num_classes
+
 
 def load_Higgs(data_path, **kwargs):
     """