fixed typo

Author: sot176

CollaborativeCoding/dataloaders/uspsh5_7_9.py

@@ -45,7 +45,7 @@ def __init__(
             Path to the USPS `.h5` file.
             
         transform : callable, optional, default=None
-            A t2ransform function to apply on images.
+            A transform function to apply on images.
         """
         self.filename = "usps.h5"
         path = data_path if isinstance(data_path, Path) else Path(data_path)

CollaborativeCoding/metrics/F1.py

@@ -5,8 +5,8 @@
 
 class F1Score(nn.Module):
     """
-    Computes Micro or Macro F1 Score efficiently.
-
+    F1 Score implementation with support for both macro and micro averaging.
+    This class computes the F1 score during training using either macro or micro averaging.
     Parameters
     ----------
     num_classes : int