输出模型测试集预测结果

baiyyang / medical-diagnosis-cnn-rnn-rcnn

分别使用rnn/cnn/rcnn来实现根据患者描述，进行疾病诊断

Apache License 2.0

52 stars 13 forks source link

Open superguopeng opened 5 years ago

superguopeng commented 5 years ago

您好，想请教下，在使用MF-TextCNN或者AF-TextCNN模型的时候，如何使用将测试集预测结果输出呢？而不是整体输出准确率？还望大神赐教

baiyyang commented 5 years ago

Hi，tf.argmax(self.input_y, 1)，这个就是获取概率最高的label，对应的是预测结果。

superguopeng commented 5 years ago

您好，这段代码加在哪里会比较合适呢？我试了几个地方，似乎都不行

baiyyang commented 5 years ago

模型训练好了，就可以直接测试了，放在训练之后的哪里都可以