diff --git a/code/compute_hyperparameters.py b/code/compute_hyperparameters.py index 56c1530ea33c45b4c30058e0f175a65b6afea071..7886bc67439c6a5fbf4df21edc78f92f444a3b48 100644 --- a/code/compute_hyperparameters.py +++ b/code/compute_hyperparameters.py @@ -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])]