Skip to content

Commit d63a0ec

Browse files
author
Ganyu Teng
committed
Add a description of the setting argument in the parser
1 parent 0f21027 commit d63a0ec

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

evaluate_anollm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def get_args():
1818
parser.add_argument("--dataset", type = str, default='wine', choices = [d.lower() for d in DATA_MAP.keys()],
1919
help="Name of datasets in the ODDS benchmark")
2020
parser.add_argument("--exp_dir", type = str, default=None)
21-
parser.add_argument("--setting", type = str, default='semi_supervised', choices = ['semi_supervised', 'unsupervised'])
21+
parser.add_argument("--setting", type = str, default='semi_supervised', choices = ['semi_supervised', 'unsupervised'], help="semi_supervised:an uncontaminated, unsupervised setting; unsupervised:a contaminated, unsupervised setting")
2222

2323
#dataset hyperparameters
2424
parser.add_argument("--data_dir", type = str, default='data')

evaluate_baselines.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def get_args():
3636
parser.add_argument("--dataset", type = str, default='wine', choices = [d.lower() for d in DATA_MAP.keys()],
3737
help="Name of datasets in the ODDS benchmark")
3838
parser.add_argument("--exp_dir", type = str, default=None)
39-
parser.add_argument("--setting", type = str, default='semi_supervised', choices = ['semi_supervised', 'unsupervised'])
39+
parser.add_argument("--setting", type = str, default='semi_supervised', choices = ['semi_supervised', 'unsupervised'], help="semi_supervised:an uncontaminated, unsupervised setting; unsupervised:a contaminated, unsupervised setting")
4040
parser.add_argument("--normalize", action='store_true', default=False) #normalize the numerical features to have zero mean and unit variance
4141
parser.add_argument("--text_encoding", type = str, default='word2vec', choices = ['tfidf', 'word2vec', 'bag_of_words'])
4242
#dataset hyperparameters

train_anollm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def get_args():
1818
parser.add_argument("--dataset", type = str, default='wine', choices = [d.lower() for d in DATA_MAP.keys()],
1919
help="Name of datasets in the ODDS benchmark")
2020
parser.add_argument("--exp_dir", type = str, default=None)
21-
parser.add_argument("--setting", type = str, default='semi_supervised', choices = ['semi_supervised', 'unsupervised'])
21+
parser.add_argument("--setting", type = str, default='semi_supervised', choices = ['semi_supervised', 'unsupervised'], help="semi_supervised:an uncontaminated, unsupervised setting; unsupervised:a contaminated, unsupervised setting")
2222

2323
# wandb
2424
parser.add_argument("--wandb", action='store_true')

0 commit comments

Comments
 (0)