7

私は pybrain を利用して、6 つの入力次元と 1 つの実数値出力次元を持つネットワークを構築しています。私が使用するコードを以下に示します。

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 配列として取得したいと考えています。エラーとともに予測結果を表示できますが、配列として表示したいです。とにかくこれができることはありますか？

4

1 に答える 1

4

ネットワークの機能を使用し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])

(XORDataSet が必要なため、pybrain のディレクトリ pybrain/examples/supervised/backprop でのみ機能します。)

于 2013-04-22T08:28:41.673 に答える