nikos99n commited on
Commit
e9624bc
·
verified ·
1 Parent(s): 5adf67c

fix syntax error

Browse files
Files changed (1) hide show
  1. src/data.py +6 -10
src/data.py CHANGED
@@ -6,13 +6,11 @@ from . import config
6
 
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
- """
16
  print("Balancing dataset (Upsampling)...")
17
 
18
  # 1. Find the maximum class count
@@ -43,9 +41,7 @@ def balance_dataset(df):
43
 
44
 
45
  def load_metadata(limit=None, balance=True):
46
- """
47
- Loads CSV, parses classes, prepares file paths, and optionally balances data.
48
- """
49
  print("Loading Metadata...")
50
  if not os.path.exists(config.CSV_PATH):
51
  raise FileNotFoundError(f"CSV not found at {config.CSV_PATH}")
 
6
 
7
 
8
  def balance_dataset(df):
9
+ #NOT USED CURRENTLY because we augment the dataset.
10
+
11
+ #Upsamples minority classes to match the count of the majority class.
12
+ #This ensures the model sees an equal number of examples for each lesion type.
13
+
 
 
14
  print("Balancing dataset (Upsampling)...")
15
 
16
  # 1. Find the maximum class count
 
41
 
42
 
43
  def load_metadata(limit=None, balance=True):
44
+ #Loads CSV, parses classes, prepares file paths, and optionally balances data.
 
 
45
  print("Loading Metadata...")
46
  if not os.path.exists(config.CSV_PATH):
47
  raise FileNotFoundError(f"CSV not found at {config.CSV_PATH}")