nikos99n commited on
Commit
ba046d2
·
verified ·
1 Parent(s): ce9c417

Update src/data.py

Browse files
Files changed (1) hide show
  1. src/data.py +3 -0
src/data.py CHANGED
@@ -7,6 +7,9 @@ from . import config
7
 
8
  def balance_dataset(df):
9
  """
 
 
 
10
  Upsamples minority classes to match the count of the majority class.
11
  This ensures the model sees an equal number of examples for each lesion type.
12
  """
 
7
 
8
  def balance_dataset(df):
9
  """
10
+ NOT USED CURRENTLY because we augment the dataset.
11
+ """
12
+
13
  Upsamples minority classes to match the count of the majority class.
14
  This ensures the model sees an equal number of examples for each lesion type.
15
  """