【发布时间】:2020-02-07 21:19:01
【问题描述】:
我是深度学习和 Keras 的完整初学者。我想建立一个分层的注意力网络,帮助将 cmets 分类为几个类别,即。有毒,剧毒等。我从一个开放的存储库中获取了代码并保存了模型。然后我使用 model_from_json 加载模型。现在我希望使用这个加载的模型对输入文本进行预测(作为 python 输入或单独的文件给出)。
这是我正在使用的代码:https://www.kaggle.com/sermakarevich/hierarchical-attention-network/notebook
然后我做了:
model_json = model.to_json()
with open("model.json", "w") as json_file:
json_file.write(model_json)
model.save_weights("model.h5")
print("Saved model to disk")
然后在一个单独的文件中:
json_file = open('model.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json,custom_objects={'AttentionWithContext':AttentionWithContext})
loaded_model.load_weights("model.h5")
print("Loaded model from disk")
我正在完美地“从磁盘加载模型”。我想知道我需要输入的格式以及如何使用代码 sn-p 使用模型对其进行分类。由于我对此知之甚少,如果有人可以帮助我使用 python 特定代码使其工作,那将非常有帮助。
【问题讨论】:
标签: machine-learning keras deep-learning data-science attention-model