如何在pybrain中保存动作表

Question

我想使用Pybrain强化学习来构建一个玩游戏的人工智能。我使用Reinforcement Learning，它会生成一个动作值表。但我不知道如何保存它，并在我训练它之后阅读。

我已经在pybrain中找到了如何保存网络的方法，但它不能保存动作值表。

from pybrain.tools.shortcuts import buildNetwork
from pybrain.tools.customxml import NetworkWriter
from pybrain.tools.customxml import NetworkReader

net = buildNetwork(4,6,1)

NetworkWriter.writeToFile(net, 'filename.xml')
net = NetworkReader.readFrom('filename.xml')

Answer 1

只需使用.getActionValues(<row>)方法：

from __future__ import print_function
from pybrain.rl.learners.valuebased import ActionValueTable

a = ActionValueTable(5, 5)
a.initialize(0.)

table = []
for i in range(a.numRows):
    table.append(a.getActionValues(i))

print(str(table))