def parse_options():
options = dict()
options['config'] = 'config/bimodal.ini'
options['no_plot'] = False
parser = argparse.ArgumentParser()
parser.add_argument('--config', help='config file to use, default=config/bimodal.ini')
parser.add_argument('--write_results', help='write results to file')
parser.add_argument('--update_rule', help='adadelta, sgdm, sgdnm, adam')
parser.add_argument('--learning_rate', help='learning rate')
parser.add_argument('--decay_rate', help='learning rate decay')
parser.add_argument('--momentum', help='momentum')
parser.add_argument('--momentum_schedule', help='eg: 0.9,0.9,0.95,0.99')
parser.add_argument('--validation_window', help='validation window length, eg: 6')
parser.add_argument('--t1', help='epoch to start learning rate decay, eg: 10')
parser.add_argument('--weight_init', help='norm,glorot,ortho,uniform')
parser.add_argument('--num_epoch', help='number of epochs to run')
parser.add_argument('--use_peepholes', help='use peephole connections in LSTM')
parser.add_argument('--no_plot', dest='no_plot', action='store_true', help='disable plots')
parser.set_defaults(no_plot=False)
parser.set_defaults(use_peepholes=False)
args = parser.parse_args()
if args.config:
options['config'] = args.config
if args.write_results:
options['write_results'] = args.write_results
if args.update_rule:
options['update_rule'] = args.update_rule
if args.learning_rate:
options['learning_rate'] = args.learning_rate
if args.decay_rate:
options['decay_rate'] = args.decay_rate
if args.momentum:
options['momentum'] = args.momentum
if args.momentum_schedule:
options['momentum_schedule'] = args.momentum_schedule
if args.validation_window:
options['validation_window'] = args.validation_window
if args.t1:
options['t1'] = args.t1
if args.weight_init:
options['weight_init'] = args.weight_init
if args.num_epoch:
options['num_epoch'] = args.num_epoch
if args.no_plot:
options['no_plot'] = True
if args.use_peepholes:
options['use_peepholes'] = True
return options
评论列表
文章目录
