auto print dataset.stat()

sunlabuiuc · Nov 15, 2022 · bc78800 · bc78800
1 parent d3bf727
commit bc78800
Show file tree

Hide file tree

Showing 9 changed files with 14 additions and 14 deletions.
diff --git a/examples/drug_recommendation_mimic3_gamenet.py b/examples/drug_recommendation_mimic3_gamenet.py
@@ -10,11 +10,11 @@
  tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
  code_mapping={"NDC": ("ATC", {"target_kwargs": {"level": 3}})},
 )
-print(dataset.stat())
+dataset.stat()
 
 # STEP 2: set task
 dataset.set_task(drug_recommendation_mimic3_fn)
-print(dataset.stat())
+dataset.stat()
 
 train_dataset, val_dataset, test_dataset = split_by_patient(dataset, [0.8, 0.1, 0.1])
 train_dataloader = get_dataloader(train_dataset, batch_size=32, shuffle=True)

diff --git a/examples/drug_recommendation_mimic3_transformer.py b/examples/drug_recommendation_mimic3_transformer.py
@@ -10,11 +10,11 @@
  tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
  code_mapping={"NDC": ("ATC", {"target_kwargs": {"level": 3}})},
 )
-print(dataset.stat())
+dataset.stat()
 
 # STEP 2: set task
 dataset.set_task(drug_recommendation_mimic3_fn)
-print(dataset.stat())
+dataset.stat()
 
 train_dataset, val_dataset, test_dataset = split_by_patient(dataset, [0.8, 0.1, 0.1])
 train_dataloader = get_dataloader(train_dataset, batch_size=32, shuffle=True)

diff --git a/examples/length_of_stay_mimic3_rnn.py b/examples/length_of_stay_mimic3_rnn.py
@@ -10,11 +10,11 @@
  tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
  code_mapping={"ICD9CM": "CCSCM", "ICD9PROC": "CCSPROC", "NDC": "ATC"},
 )
-print(dataset.stat())
+dataset.stat()
 
 # STEP 2: set task
 dataset.set_task(length_of_stay_prediction_mimic3_fn)
-print(dataset.stat())
+dataset.stat()
 
 train_dataset, val_dataset, test_dataset = split_by_patient(dataset, [0.8, 0.1, 0.1])
 train_dataloader = get_dataloader(train_dataset, batch_size=32, shuffle=True)

diff --git a/examples/mortality_mimic3_rnn.py b/examples/mortality_mimic3_rnn.py
@@ -10,11 +10,11 @@
  tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
  code_mapping={"ICD9CM": "CCSCM", "ICD9PROC": "CCSPROC", "NDC": "ATC"},
 )
-print(dataset.stat())
+dataset.stat()
 
 # STEP 2: set task
 dataset.set_task(mortality_prediction_mimic3_fn)
-print(dataset.stat())
+dataset.stat()
 
 train_dataset, val_dataset, test_dataset = split_by_patient(dataset, [0.8, 0.1, 0.1])
 train_dataloader = get_dataloader(train_dataset, batch_size=32, shuffle=True)

diff --git a/examples/readmission_mimic3_rnn.py b/examples/readmission_mimic3_rnn.py
@@ -10,11 +10,11 @@
  tables=["DIAGNOSES_ICD", "PROCEDURES_ICD", "PRESCRIPTIONS"],
  code_mapping={"ICD9CM": "CCSCM", "ICD9PROC": "CCSPROC", "NDC": "ATC"},
 )
-print(dataset.stat())
+dataset.stat()
 
 # STEP 2: set task
 dataset.set_task(readmission_prediction_mimic3_fn)
-print(dataset.stat())
+dataset.stat()
 
 train_dataset, val_dataset, test_dataset = split_by_patient(dataset, [0.8, 0.1, 0.1])
 train_dataloader = get_dataloader(train_dataset, batch_size=32, shuffle=True)

diff --git a/pyhealth/datasets/eicu.py b/pyhealth/datasets/eicu.py
@@ -490,5 +490,5 @@ def parse_physicalexam(self, patients: Dict[str, Patient]) -> Dict[str, Patient]
  dev=False,
  refresh_cache=True,
  )
- print(dataset.stat())
+ dataset.stat()
  dataset.info()
diff --git a/pyhealth/datasets/mimic3.py b/pyhealth/datasets/mimic3.py
@@ -339,7 +339,7 @@ def parse_labevents(self, patients: Dict[str, Patient]) -> Dict[str, Patient]:
  code_mapping={"NDC": "ATC"},
  refresh_cache=True,
  )
- print(dataset.stat())
+ dataset.stat()
  dataset.info()
 
  # dataset = MIMIC3Dataset(

diff --git a/pyhealth/datasets/mimic4.py b/pyhealth/datasets/mimic4.py
@@ -349,5 +349,5 @@ def parse_labevents(self, patients: Dict[str, Patient]) -> Dict[str, Patient]:
  code_mapping={"NDC": "ATC"},
  refresh_cache=True,
  )
- print(dataset.stat())
+ dataset.stat()
  dataset.info()
diff --git a/pyhealth/datasets/omop.py b/pyhealth/datasets/omop.py
@@ -412,5 +412,5 @@ def parse_measurement(self, patients: Dict[str, Patient]) -> Dict[str, Patient]:
  dev=False,
  refresh_cache=True,
  )
- print(dataset.stat())
+ dataset.stat()
  dataset.info()