示例#1
0
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)
示例#2
0
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)
示例#3
0
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)