lishimin
9a0a7d1076
上传文件至 'data'
|
5 years ago | |
|---|---|---|
| data | 5 years ago | |
| pickle_1 | 5 years ago | |
| README.md | 5 years ago | |
| bidi_classify_stop_words.csv | 5 years ago | |
| data_precess.py | 5 years ago | |
| data_util.py | 5 years ago | |
| main.py | 5 years ago | |
| padded_sequences_te.pkl | 5 years ago | |
| test_label.pkl | 5 years ago |
README.md
1、标注数据： data/train_11.csv # 训练数据 data/test_11.csv # 测试数据
2、其他重要文件： pickle_1/word_index_955871.pk #词典文件 word:id pickle_1/tokenizer_955871.pk # 训练后keras tokenizer对象 model/thr_100_model.vector # 词向量文件 pickle_1/label_mapping_f.pk # 标签编号字典 pickle_1/class_subclass_dic211.pk'# 文章类别字典
3、最终保存模型文件： model/New_attentionLSTM_weights1_100_em21.h5
4、模型及主要代码在main.py 文件中 ，直接train() /test() 进行训练或测试即可