1.原始神经网络cnn+bilstm训练好模型,然后预测做命名实体识别。
from ner import Parser
def getChouquMail(request):
if request.method == 'POST':
p = Parser()
p.load_models("models/") #训练的模型存在models文 件下面
str="你需要命名实体识别的部分"
a = p.predict(str)
print(a)
就能输出想要的结果,但是结合django模型就会报错。推荐参考文档https://blog.csdn.net/weixin_30034903/article/details/106144194
https://blog.csdn.net/xxzhix/article/details/81983982
问题修改描述:
1.一开始出现tensorflow多次restore模型出错。然后自己添加 K.clear_session()代码。
2.用keras训练好了文本分类的模型,用h5格式进行保存,再用django中加载这个模型。在django初始化加载模型的过程是没问题的,但是一旦调用函数使用模型时执行model.predict就会报错:Tensor Tensor(" …dtype=float32") is not element of this graph
2.修改后
from ner import Parser
from keras import backend as K
import tensorflow as tf
from keras.models import load_model
graph = tf.get_default_graph()
model = load_model('models/model.h5')
def getChouquMail(request):
global model
global graph
K.clear_session()
with graph.as_default():
if request.method == 'POST':
p = Parser()
p.load_models("models/")
str = "你需要命名实体识别的部分"
a = p.predict(str)
print(a)