将pybrain预测的输出作为数组获取

【问题标题】：Getting output of pybrain prediction as array将pybrain预测的输出作为数组获取
【发布时间】：2013-04-15 00:13:33
【问题描述】：

我正在使用 pybrain 构建一个具有 6 个输入维度和一个实值输出维度的网络。我使用的代码如下所示：

network = buildNetwork(train.indim, 4, train.outdim)

trainer = BackpropTrainer( network, train)
trainer.trainOnDataset(train, 8000)

print 'MSE train', trainer.testOnData(train, verbose = True)

这里的火车是数据集类型我想将 trainer.testOnData() 中的预测作为一个 numpy 数组。我能够查看预测结果以及错误，但我希望它作为一个数组。无论如何，这可以做到吗？

【问题讨论】：

标签： python numpy machine-learning neural-network pybrain

【解决方案1】：

使用您网络的activate 功能：

numpy.array([network.activate(x) for x, _ in train])

完整示例：

from datasets import XORDataSet 
from pybrain.tools.shortcuts import buildNetwork
from pybrain.supervised import BackpropTrainer
import numpy
d = XORDataSet()
n = buildNetwork(d.indim, 4, d.outdim, bias=True)
t = BackpropTrainer(n, learningrate=0.01, momentum=0.99, verbose=True)
t.trainOnDataset(d, 1000)
t.testOnData(verbose=True)
print numpy.array([n.activate(x) for x, _ in d])

（只适用于pybrain的目录pybrain/examples/supervised/backprop，因为需要XORDataSet。）

【讨论】：