Spaces:

svalcin
/

MultiplyGame

Sleeping

svalcin commited on Sep 9, 2023

Commit

2416c46

•

1 Parent(s): 5f37c00

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ class QModel():
     #returns the top n states
     def predict(self,state_n):
         #return np.argsort(self.qtable[state_n][:])[-n:]
-        np.argmax(self.qtable[state_n][:])
 learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))

     #returns the top n states
     def predict(self,state_n):
         #return np.argsort(self.qtable[state_n][:])[-n:]
+        np.argmax(self.qtable[state_n])
 learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))