使用Theano CNN获取NLP的输出类概率

时间:2016-10-29 20:30:18

标签: python nlp theano

我正在尝试使用在Theano中使用MLPDrouput的分类器将推文分类为K类别中的一个。 虽然我能够获得给定测试推文所属的最终类,但我更感兴趣的是获得给定推文的单个类概率。例如。如果我有3个类,那么我想得到类概率值,如[0.4,0.14,0.8],基于我的分类器进行最终的输出决策。这是我的分类器预测的代码片段:

for conv_layer in conv_layers:
    test_layer0_output = conv_layer.predict(test_layer0_input, test_size)
    test_pred_layers.append(test_layer0_output.flatten(2))
test_layer1_input = T.concatenate(test_pred_layers, 1)
# test_y_pred to store the result of each tweet
test_y_pred = classifier.predict(test_layer1_input)

迭代 test_y_pred 给了我最终的输出类,但我无法找到获得prbabilities的方法。有人可以建议获得输出类概率的方法吗?

0 个答案:

没有答案