mm/mud-games
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix typos

Browse Source
This commit is contained in:
Michael Pilosov 2022-03-20 19:32:15 -06:00
parent dc353276d7
commit 6020752d75
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.py
View File

@ -5,6 +5,7 @@ import gym
import numpy as np
import pandas as pd
from scipy.stats import gaussian_kde as gkde
from scipy.stats import norm
def train(data):
@ -24,7 +25,7 @@ def train(data):
def test(decision=np.array([-0.09, -0.71, -0.43 , -0.74]), seed=1992):
env = gym.make("CartPole-v1")
observation, info = env.reset(seed=seed, return_info=True)
score = 1
score = 0
for i in range(10000):
action = 1 if decision.T @ observation < 0 else 0
observation, reward, done, info = env.step(action)
@ -34,8 +35,8 @@ def test(decision=np.array([-0.09, -0.71, -0.43 , -0.74]), seed=1992):
if score == 500:
print("WIN")
else:
print("LOSE: {int(score)}")
score = 1 # reset score
print(f"LOSE: {int(score)}")
score = 0 # reset score
observation, info = env.reset(return_info=True)
env.close()
@ -43,5 +44,5 @@ def test(decision=np.array([-0.09, -0.71, -0.43 , -0.74]), seed=1992):
if __name__ == "__main__":
data = pickle.load(open('data.pkl','rb'))
mud_point = train(data)
print("MUD Point: {mud_point}")
print(f"MUD Point: {mud_point}")
test(mud_point)