Yeefei commited on
Commit
9ecb0d4
·
verified ·
1 Parent(s): a270942

Upload datasets.py

Browse files
Files changed (1) hide show
  1. datasets.py +1 -1
datasets.py CHANGED
@@ -370,7 +370,7 @@ class MIMICDataset(Dataset):
370
  elif disease == "Lung Opacity":
371
  finding = 2
372
  else:
373
- finding = None
374
 
375
 
376
  self.samples["x"].append(img_path)
 
370
  elif disease == "Lung Opacity":
371
  finding = 2
372
  else:
373
+ finding = 0
374
 
375
 
376
  self.samples["x"].append(img_path)