def main(sequences, playerNameArg, numItersArg, sizeArg, debugArg, depthArg,
probCutoffArg, trainArg, loadArg, saveArg):
"""Primary test harness."""
startTime = time.time()
for i, sequence in enumerate(sequences):
# setup parameters for sequence
params = generateSequenceParams(sequence, [
('playerName', playerNameArg),
('numIters', numItersArg),
('size', sizeArg),
('debug', debugArg),
('depth', depthArg),
('probCutoff', probCutoffArg),
('train', trainArg),
('load', loadArg),
('save', saveArg),
])
print "Running sequence %d: %s\n" % (i + 1, sequence)
print "Full parameters: %s\n" % params
Model.Model.SIZE = params['size']
if params['playerName'] == PLAYER_NAMES[0]:
# run the interactive player
p = Player.InteractivePlayer(debug=params['debug'])
p.run(numIters=params['numIters'])
elif params['playerName'] == PLAYER_NAMES[1]:
# run the baseline greedy player
p = Player.BaselineGreedyPlayer(debug=params['debug'])
p.run(numIters=params['numIters'],
printStats=True,
printAtCheckpoints=True)
elif params['playerName'] == PLAYER_NAMES[2]:
# run the baseline corner player
p = Player.BaselineCornerPlayer(debug=params['debug'])
p.run(numIters=params['numIters'],
printStats=True,
printAtCheckpoints=True)
elif params['playerName'] == PLAYER_NAMES[3]:
# run the baseline random player
p = Player.BaselineRandomPlayer(debug=params['debug'])
p.run(numIters=params['numIters'],
printStats=True,
printAtCheckpoints=True)
elif params['playerName'] == PLAYER_NAMES[4]:
# run the expectimax player
p = ExpectiMaxPlayer.ExpectiMaxPlayer(
debug=params['debug'],
depth=params['depth'],
probCutoff=params['probCutoff'])
p.run(numIters=params['numIters'],
printStats=True,
printAtCheckpoints=True)
elif params['playerName'] == PLAYER_NAMES[5]:
# run the RL player
p = QLPlayer.RLPlayer(debug=params['debug'],
train=params['train'],
load=params['load'],
save=params['save'])
p.run(numIters=params['numIters'],
printStats=True,
printAtCheckpoints=True)
runTime = time.time() - startTime
print "\nTotal runtime after %d sequence(s): %.3f seconds" % (i + 1,
runTime)
if i != len(sequences) - 1:
print "\n\n"
def setUp(self):
self.p = Player.BaselineCornerPlayer()