Ver código fonte

first commit

lishimin 5 anos atrás
commit
5a122130d9
1 arquivos alterados com 15 adições e 0 exclusões
  1. 15 0
      README.md

+ 15 - 0
README.md

@@ -0,0 +1,15 @@
+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() 进行训练或测试即可