Commit 91953d7f authored by Steven Cordwell's avatar Steven Cordwell
Browse files

fix not running mdp

parent c205c76f
# -*- 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)
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