Skip to content
Snippets Groups Projects
Commit 7494dbc4 authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Rename 'use_different_seed_number' parameter to 'use_variable_seed_number'

parent e319343d
No related branches found
No related tags found
1 merge request!9Resolve "Experiment pipeline"
......@@ -135,14 +135,14 @@ if __name__ == "__main__":
'max_features': Categorical(['auto', 'sqrt', 'log2'], [0.5, 0.25, 0.25])}
begin_random_seed_range = 1
end_random_seed_range = 2000
DEFAULT_USE_DIFFERENT_SEED_NUMBER = False
DEFAULT_USE_VARIABLE_SEED_NUMBER = False
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--cv', nargs='?', type=int, default=DEFAULT_CV, help='Specify the size of the cross-validation.')
parser.add_argument('--n_iter', nargs='?', type=int, default=DEFAULT_N_ITER, help='Specify the number of iterations for the bayesian search.')
parser.add_argument('--random_seed_number', nargs='?', type=int, default=DatasetLoader.DEFAULT_RANDOM_SEED_NUMBER, help='Number of random seeds used.')
parser.add_argument('--seeds', nargs='+', type=int, default=None, help='Specific a list of seeds instead of generate them randomly')
parser.add_argument('--use_different_seed_number', action='store_true', default=DEFAULT_USE_DIFFERENT_SEED_NUMBER, help='Compute the amount of random seeds depending on the dataset.')
parser.add_argument('--use_variable_seed_number', action='store_true', default=DEFAULT_USE_VARIABLE_SEED_NUMBER, help='Compute the amount of random seeds depending on the dataset.')
parser.add_argument('--datasets', nargs='+', type=str, default=DatasetLoader.dataset_names, help='Specify the dataset used by the estimator.')
parser.add_argument('--verbose', action='store_true', default=DEFAULT_VERBOSE, help='Print information during the bayesian search.')
args = parser.parse_args()
......@@ -153,7 +153,7 @@ if __name__ == "__main__":
logger.warning('seeds and random_seed_number parameters are both specified. Seeds will be used.')
# Seeds are either provided as parameters or generated at random
if args.use_different_seed_number:
if args.use_variable_seed_number:
seeds = args.seeds if args.seeds is not None \
else [random.randint(begin_random_seed_range, end_random_seed_range) \
for i in range(args.random_seed_number)]
......@@ -169,7 +169,7 @@ if __name__ == "__main__":
DatasetLoader.dataset_seed_numbers dictionary, depending on
the dataset.
"""
if args.use_different_seed_number:
if args.use_variable_seed_number:
seeds = [random.randint(begin_random_seed_range, end_random_seed_range) \
for i in range(DatasetLoader.dataset_seed_numbers[dataset_name])]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment