关于是否能输出分类匹配后的标签
yjy4201227 opened this issue · comments
您好！再次打扰您！之前跑通代码后得到的是
R8:
Test set results: cost= 0.11023 accuracy= 0.97168 time= 1.06711
15362
Test Precision, Recall and F1-Score...
现在我想问一下是如何得到论文中表格展示的**“20NG中几个类别最高值的单词。使用第二层词嵌入。为每个类别展示前10个单词”**
中途是否有分类标签输出展示每个测试集是否匹配呢
谢谢您！
训练的时候，把train.py这里的0，改成1, 可以得到第二层所有节点的embedding:
outs = sess.run([model.opt_op, model.loss, model.accuracy,
model.layers[0].embedding], feed_dict=feed_dict)
这个embedding跟labels的维度是一样的，对应关系build_graph.py里面有
训练的时候，把train.py这里的0，改成1, 可以得到第二层所有节点的embedding:
outs = sess.run([model.opt_op, model.loss, model.accuracy, model.layers[0].embedding], feed_dict=feed_dict)
这个embedding跟labels的维度是一样的，对应关系build_graph.py里面有
好的谢谢您！我后面自己修改运行一下！