- Added the LUNG_CANCER_CT as target dataset

franck-armand · Nov 18, 2024 · 09056b7 · 09056b7
1 parent 6dc2e9a
commit 09056b7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/params.py b/params.py
@@ -23,7 +23,13 @@
 src_model_trained = True
 
 # params for target dataset
-tgt_dataset = "USPS"
+# tgt_dataset = "USPS"
+# tgt_encoder_restore = "snapshots/ADDA-target-encoder-final.pt"
+# tgt_model_trained = True
+
+# params for target dataset lung cqncer ct scan 
+tgt_dataset = "LUNG_CANCER_CT"
+tgt_dataset_path = r"data/lung_cancer_ct_scan_train.csv"
 tgt_encoder_restore = "snapshots/ADDA-target-encoder-final.pt"
 tgt_model_trained = True