Changes label

microsoft · Dec 2, 2021 · 3de388b · 3de388b
1 parent 56b32ac
commit 3de388b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/InnerEye/ML/Histopathology/datasets/tcga_prad_dataset.py b/InnerEye/ML/Histopathology/datasets/tcga_prad_dataset.py
@@ -12,12 +12,12 @@ class TcgaPradDataset(Dataset):
     - `'slide_id'` (str)
     - `'case_id'` (str)
     - `'image_path'` (str): absolute slide image path
-    - `'brca12'` (int, 0 or 1): label for predicting BRCA1/2 mutations
+    - `'label'` (int, 0 or 1): label for predicting positive or negative
     """
     SLIDE_ID_COLUMN: str = 'slide_id'
     CASE_ID_COLUMN: str = 'case_id'
     IMAGE_COLUMN: str = 'image_path'
-    LABEL_COLUMN: str = 'brca12'
+    LABEL_COLUMN: str = 'label'
 
     DEFAULT_CSV_FILENAME: str = "dataset.csv"
 
@@ -40,8 +40,8 @@ def __init__(self, root_dir: Union[str, Path],
             dataset_df = pd.read_csv(self.dataset_csv)
 
         dataset_df = dataset_df.set_index(self.SLIDE_ID_COLUMN)
-        dataset_df[self.LABEL_COLUMN] = (dataset_df['brca1_mutation']
-                                         | dataset_df['brca2_mutation']).astype(int)
+        dataset_df[self.LABEL_COLUMN] = (dataset_df['label1_mutation']
+                                         | dataset_df['label2_mutation']).astype(int)
         self.dataset_df = dataset_df
 
     def __len__(self) -> int: