EBIM-NLI
这是Keras中自然语言推理的增强BiLSTM推理模型的实现。 该模型基于Chen等人的论文。 链接:
使用的数据集是。 该模型使用预先训练的单词向量。
用法
需要Keras训练模型并进行测试。 需要执行NLI.ipynb来训练和测试模型。
训练模型后,将生成以下文件:
1. tokenizer.pickle - tokenizes sentences
2. embeddings.npy - Word embeddings based on the GloVe model.
3. NLI.h5 - trained weights for the EBIM model
app.py使用这些文件来预测给定输入的类别。