def manage_model_list(): trainset = request.args.get('trainset', None) config = request.args.get('config', None) model_dir = get_model_dir(trainset, config) file_list = get_model_list(model_dir) print(file_list) form = SelectEvalForm() form.select_model.choices = get_options(file_list) form.select_test.choices = get_options(get_files()) if request.method == 'POST': print("form:", request.form) data_choices = dict(get_options(get_files())) model_choices = dict(get_options(file_list)) print(get_configs()) select_model = model_choices.get(form.select_model.data) select_test = data_choices.get(form.select_test.data) print('model', select_model) print('test:', select_test) return redirect( url_for('eval_model', trainname=trainset, config=config, testname=select_test, modelname=select_model)) return render_template('model_download.html', form=form, trainname=trainset, config=config, files_list=file_list)
def manage_job(error_message): print('jobs') print(app.task_queue.job_ids) dict_status = get_file_status() dict_valid_status = get_valid_status() print(app.job_file2id) print(dict_status) form = SelectConfigForm() form.select_config.choices = get_options(get_configs()) form.select_data.choices = get_options(get_files()) if request.method == 'POST': print("form:", request.form) data_choices = dict(get_options(get_files())) config_choices = dict(get_options(get_configs())) print(get_configs()) select_config = config_choices.get(form.select_config.data) select_data = data_choices.get(form.select_data.data) print('data', select_data) print('config:', select_config) if form.submit.data: return redirect( url_for('train_model', filename=select_data, config=select_config)) elif form.submit_valid.data: return redirect( url_for('valid_model', filename=select_data, config=select_config)) return render_template('jobs.html', form=form, dict_status=dict_status, dict_valid_status=dict_valid_status, errors=error_message)
def manage_data(): files_list = get_files() form = UploadDataForm() f = form.archive.data if form.validate_on_submit(): filename = secure_filename(f.filename) prefix = '.'.join(filename.rsplit('.', 2)[:-2]) filename = rename_file(prefix, '.tar.gz', files_list) f.save(os.path.join(app.config['UPLOAD_FOLDER'], filename)) return redirect(url_for('manage_data')) return render_template('data.html', form=form, files_list=files_list)