def run():
parser = GooeyParser()
subs = parser.add_subparsers(help='commands', dest='commands')
train_parser = subs.add_parser('train', help='Configurate model training')
param_group = train_parser.add_argument_group("Model parameter option", gooey_options={'show_border': True, 'columns': 2})
args_param.add(param_group)
data_group = train_parser.add_argument_group("Data Options", gooey_options={'show_border': True}, )
args_data.add(data_group)
save_group = train_parser.add_argument_group("Save option", gooey_options={'show_border': True, 'columns': 2})
args_save.add(save_group)
test_parser = subs.add_parser('test', help='Configurate model testining')
data_group = test_parser.add_argument_group("Data Options", gooey_options={'show_border': True}, )
args_data.add(data_group)
load_group = test_parser.add_argument_group("Load option", gooey_options={'show_border': True, 'columns': 1})
args_load.add(load_group, model_savefiles())
save_group = test_parser.add_argument_group("Save option", gooey_options={'show_border': True, 'columns': 2})
args_save.add(save_group)
args = parser.parse_args()
X, Y = load_data(args.data_path)
if args.commands =='train':
train(args, X, Y, save_dir)
else:
with open(save_dir + args.load_model, 'rb') as f:
model = pickle.load(f)
test(args, X, Y, save_dir, model)
def run():
parser = GooeyParser()
subs = parser.add_subparsers(help='commands', dest='commands')
train_parser = subs.add_parser('train', help='Configurate model training')
param_group = train_parser.add_argument_group("Model parameter option",
gooey_options={
'show_border': True,
'columns': 2
})
args_param.add(param_group)
data_group = train_parser.add_argument_group("Data Options",
gooey_options={
'show_border': True,
'columns': 1
})
args_data.add(data_group)
save_group = train_parser.add_argument_group("Save option",
gooey_options={
'show_border': True,
'columns': 1
})
args_save.add(save_group)
test_parser = subs.add_parser('test', help='Configurate model testing')
data_group = test_parser.add_argument_group("Data Options",
gooey_options={
'show_border': True,
'columns': 1
})
args_data.add(data_group)
pretrained_group = test_parser.add_argument_group(
"Pretrained file Options",
gooey_options={'show_border': True},
)
pretrained_group = args_pre.add(pretrained_group)
save_group = test_parser.add_argument_group("Save option",
gooey_options={
'show_border': True,
'columns': 1
})
args_save.add(save_group)
args = parser.parse_args()
if args.commands == 'train':
train.train(args)
else:
test.test(args)