2024-05-10 09:59:24 +00:00
|
|
|
import argparse
|
|
|
|
|
from argparse import ArgumentParser
|
|
|
|
|
|
|
|
|
|
# Construct an argument parser
|
|
|
|
|
all_args = argparse.ArgumentParser()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def makeArguments(arguments: ArgumentParser) -> dict:
|
|
|
|
|
all_args.add_argument("-b", "--Bayesian", action="store", dest="b",
|
|
|
|
|
type=int, choices=range(1, 8),
|
|
|
|
|
help="Bayesian model of size x")
|
|
|
|
|
all_args.add_argument("-f", "--Frequentist", action="store", dest="f",
|
|
|
|
|
type=int, choices=range(1, 8),
|
|
|
|
|
help="Frequentist model of size x")
|
|
|
|
|
all_args.add_argument("-E", "--EarlyStopping", action="store_true",
|
|
|
|
|
help="Early Stopping criteria")
|
|
|
|
|
all_args.add_argument("-e", "--EnergyBound", action="store_true",
|
|
|
|
|
help="Energy Bound criteria")
|
|
|
|
|
all_args.add_argument("-a", "--AccuracyBound", action="store_true",
|
|
|
|
|
help="Accuracy Bound criteria")
|
|
|
|
|
all_args.add_argument("-s", "--Save", action="store_true",
|
|
|
|
|
help="Save model")
|
|
|
|
|
all_args.add_argument('--net_type', default='lenet', type=str,
|
|
|
|
|
help='model = [lenet/AlexNet/3Conv3FC]')
|
2024-05-10 10:04:28 +00:00
|
|
|
all_args.add_argument('--dataset', default='MNIST', type=str,
|
2024-05-10 09:59:24 +00:00
|
|
|
help='dataset = [MNIST/CIFAR10/CIFAR100]')
|
|
|
|
|
return vars(all_args.parse_args())
|
