script.py 文件源码

def lstm_predict(filename, content) : 
    print(content)
    model_input = sequence.pad_sequences([one_hot(str(content), 5000)], 1000)
    model = load_model(filename)
    prediction = model.predict(model_input)
    return prediction.tolist()[0][0]