使用ano-CNN获取输出类概率

2021-10-17 17:46:38 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用一个分类器,在Theano中使用MLPDrouput将tweet分类为K个类别之一。 虽然我能够得到给定测试tweet所属的最后一个类,但我更感兴趣的是获得给定tweet的单个类概率。例如,如果我有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\u y\u pred会得到最终的输出类,但我无法找到获得性能的方法。有人能提出获得输出类概率的方法吗?你知道吗