Commit eb101575 authored by Steven Cordwell's avatar Steven Cordwell

Merge branch 'hotfix'

Conflicts:
	setup.py
parents 7f051c5e 91953d7f
# -*- coding: utf-8 -*-
import cPickle as pickle
import numpy as np
from scipy.sparse import dok_matrix
......@@ -181,4 +183,8 @@ def isValid(state):
if __name__ == "__main__":
P, R = getTransitionAndRewardArrays()
ttt = mdp.ValueIteration(P, R, 1)
print(ttt.policy)
ttt.setVerbose()
ttt.run()
f = "tictactoe.pkl"
pickle.dump(ttt.policy, open(f, "wb"))
print("Optimal policy pickled as '%s' in current directory." % f)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment