From 83e81722d29c185dd8cd32292510abd61bd429fd Mon Sep 17 00:00:00 2001
From: ltcptgeneral <learthurgo@gmail.com>
Date: Tue, 12 Mar 2024 22:14:03 -0700
Subject: [PATCH] this should probably be working but isn't

---
 .gitignore    |  3 ++-
 dqn_wordle.py | 24 ++++++++++++++++++++++++
 2 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 dqn_wordle.py

diff --git a/.gitignore b/.gitignore
index 8f14cb9..1244f49 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
-**/data/*
\ No newline at end of file
+**/data/*
+**/*.zip
\ No newline at end of file
diff --git a/dqn_wordle.py b/dqn_wordle.py
new file mode 100644
index 0000000..36ad429
--- /dev/null
+++ b/dqn_wordle.py
@@ -0,0 +1,24 @@
+import gym
+import gym_wordle
+from stable_baselines3 import DQN
+
+env = gym.make("Wordle-v0")
+done = False
+
+print(env)
+
+model = DQN("MlpPolicy", env, verbose=1)
+model.learn(total_timesteps=10000, log_interval=100)
+model.save("dqn_wordle")
+
+del model # remove to demonstrate saving and loading
+
+model = DQN.load("dqn_wordle")
+
+state = env.reset()
+
+while not done:
+
+    action, _states = model.predict(state, deterministic=True)
+
+    state, reward, done, info = env.step(action)
\ No newline at end of file