def upload():
form = TrainingDataForm()
if form.validate_on_submit():
model_name = form.model_name.data
learning_rate = float(form.learning_rate.data)
batch_size = int(form.batch_size.data)
filename = secure_filename(form.training_data.data.filename)
print(form.__dict__)
# Save to Redis here
form.training_data.data.save('wine_quality/data/' + filename)
dataframe = pd.read_csv('wine_quality/data/' + filename, sep=',')
my_model.train(dataframe, learning_rate, batch_size, model_name)
else:
filename = None
return render_template('test_data_upload.html', form=form, filename=filename)
评论列表
文章目录