Commit f30f2588 authored by Steven Cordwell's avatar Steven Cordwell
Browse files

fix sqlite string and passing reward into db

parent 1ffe678e
......@@ -93,6 +93,7 @@
import sqlite3
import os
from math import ceil, log, sqrt
from random import random
......@@ -545,7 +546,12 @@ def exampleRand(S, A, is_sparse=False, is_sqlite=False, mask=None):
if is_sqlite:
# to be usefully represented as a sparse matrix, the number of nonzero
# entries should be less than 1/3 of dimesion of the matrix, so (SxS)/3
conn = sqlite3.connect("transition.db")
db = "MDP.db"
if os.path.exists(db):
raise StandardError("Database already exists, not overwriting. " \
"Delete '" + db + "' if you want to continue.")
conn = sqlite3.connect(db)
with conn:
c = conn.cursor()
cmd = '''
......@@ -555,13 +561,13 @@ def exampleRand(S, A, is_sparse=False, is_sqlite=False, mask=None):
for a in range(A):
cmd = '''
CREATE TABLE transition%s (row INTEGER, "col INTEGER,
CREATE TABLE transition%s (row INTEGER, col INTEGER,
val REAL);
CREATE TABLE reward%s (val REAL);''' % (a, a)
reward = rand(S).tolist()
cmd = "INSERT INTO reward%s VALUES(?)" % a
c.executemany(cmd, reward)
c.executemany(cmd, zip(reward))
for s in xrange(S):
n = randint(1, S//3)
row = (s*ones(n, dtype=int)).tolist()
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