我正在使用 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)
这里 train 的类型是 Dataset 我想将 trainer.testOnData() 中的预测作为一个 numpy 数组。我可以查看预测结果以及错误,但我希望它作为一个数组。无论如何,这可以做到吗?