使用Flask进行机器学习部署

时间:2020-04-02 11:31:26

标签: python machine-learning flask

我正在为我的ml模型数据构建api,该数据是从HTML表单中获取的

获取错误,因为str对象没有属性'ndim' 当我在互联网上搜索此内容时,我发现这与我的字符串类型转换有关 我需要提供给我的模型。

app.py中的部分代码(第4行以下)

if request.method == 'POST':
    message = request.form['message']
    data = [message]
    data1 = str(data) //can you tell me what correct transform function in python i should use here
    my_prediction = clf.predict(data1)
return render_template('result.html',prediction = my_prediction)

我的预测功能

def gen_text(model, tokenizer, seq_len, seed_text, num_gen_words):
    output_text = []
    input_text = seed_text
    for i in range(num_gen_words):
        encoded_text = tokenizer.texts_to_sequences([input_text])[0]
        pad_encoded = pad_sequences([encoded_text], maxlen=seq_len,truncating='pre')
        pred_word_ind = model.predict_classes(pad_encoded,verbose=0)[0]

        pred_word = tokenizer.index_word[pred_word_ind]
        input_text += ' '+pred_word
        output_text.append(pred_word)
    return ' '.join(output_text)

#print('\n\n===>Enter --exit to exit from the program')

def predict(str):

    #while True:
    seed_text  = str

    out = gen_text(model, tokenizer, seq_len=seq_len, seed_text=seed_text, num_gen_words=5)
    return seed_text+' '+out
predict("aka")

0 个答案:

没有答案
相关问题