Question

我想使用Pybrain强化学习来构建AI来玩游戏。我用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))

如何在pybrain中保存动作表

1 个答案: