NLP-TupleExtraction
A final project for AI-Fundamental in UCAS 2018 Winter
Description about the file
- pretrain_data.py
预训练数据,输入为训练的.json文件,输出为处理过的.npy文件,将句子处理为 训练输入 - train.py
网络,输入为.npy文件,然后训练之后保存.ckpt - RuleMatch.py
规则匹配,可用于检查验证集的F1 - model.py
输入为测试数据.json文件,输出为.json文件,给每个sentence的dict加了一个 'results'域,'results'里面为预测的正确三元组
Reuslt
7:3划分训练集合验证集,在验证集上F1 = 0.95左右 在测试集上, precision = 0.9991, recall = 0.9682, F1 = 0.9834!