alu/cse151b-final-project
1
0
Fork 0
mirror of https://github.com/ltcptgeneral/cse151b-final-project.git synced 2024-11-10 07:04:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

try more time steps

Browse Source
This commit is contained in:
Arthur Lu 2024-03-14 10:57:18 -07:00
parent 848ea719b7
commit dd5889da33
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dqn_wordle.py
View File

@ -25,8 +25,6 @@ def test(model, env, test_num=1000):
action, _states = model.predict(obs) action, _states = model.predict(obs)
obs, rewards, done, info = env.step(action) obs, rewards, done, info = env.step(action)
print(action, obs, rewards)
return total_correct / test_num return total_correct / test_num
if __name__ == "__main__": if __name__ == "__main__":
@ -36,5 +34,5 @@ if __name__ == "__main__":
print(env) print(env)
print(model) print(model)
train(model, env, total_timesteps=10000) train(model, env, total_timesteps=500000)
print(test(model, env, test_num=1)) print(test(model, env))