Update main.py

main.py

@@ -29,8 +29,8 @@
 parser.add_argument(
     "--sparse", default=False, type=lambda x: (str(x).lower() == "true")
 )  #
-parser.add_argument("--n_trials_unsup", default=5, type=int)  #
-parser.add_argument("--n_trials_sup", default=50, type=int)  #
+parser.add_argument("--n_trials_unsup", default=20, type=int)  #
+parser.add_argument("--n_trials_sup", default=60, type=int)  #
 parser.add_argument("--alpha_masks", default="-1", type=str)  #
 parser.add_argument("--lr_alphas", default=0.001, type=float)  #
 parser.add_argument("--alpha_activation", default="none", type=str)  #
@@ -40,12 +40,6 @@
 # get options
 options = vars(parser.parse_args())
 
-if options["algorithm"] == "dgi":
-    options["augmentation_type"] = "fsgnn"
-    options["augmentation_quantity"] = 0
-    options["augmentation_index"] = 0
-    options["augmentation_all"] = False
-
 if options["algorithm"] == "mvgrl":
     options["augmentation_type"] = "diff"
     options["augmentation_quantity"] = 0
@@ -278,4 +272,4 @@ def train_supervised_wrapper(params):
 results_dict = {**unsupervised_result_dict, **supervised_result_dict}
 print("Results dict", results_dict)
 
-print("Note that these results are for the following data splits:", dataset_splits)
+print("Note that these results are for the following data splits:", dataset_splits)