Commit 9256db99 authored by Steven Cordwell's avatar Steven Cordwell
Browse files

fix exampleForest where sparse matrix did not take on user defined rewards

parent 9e4acf3b
......@@ -476,7 +476,7 @@ def exampleForest(S=3, r1=4, r2=2, p=0.1, is_sparse=False):
P = []
rows = range(S) * 2
cols = [0] * S + range(1, S) + [S - 1]
vals = [0.1] * S + [0.9] * S
vals = [p] * S + [1-p] * S
P.append(coo_matrix((vals, (rows, cols)), shape=(S,S)).tocsr())
rows = range(S)
cols = [0] * S
......
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