svalcin commited on
Commit
f82d2e5
1 Parent(s): 2416c46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ class QModel():
8
 
9
  #returns the top n states
10
  def predict(self,state_n):
11
- #return np.argsort(self.qtable[state_n][:])[-n:]
12
  np.argmax(self.qtable[state_n])
13
 
14
  learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))
@@ -18,7 +18,7 @@ def predict_next_questions(state):
18
 
19
 
20
  intf = gr.Interface(fn=predict_next_questions, inputs="number", outputs="number")
21
- print("interface created")
22
  intf.launch()
23
 
24
 
 
8
 
9
  #returns the top n states
10
  def predict(self,state_n):
11
+ print(self.qtable)
12
  np.argmax(self.qtable[state_n])
13
 
14
  learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))
 
18
 
19
 
20
  intf = gr.Interface(fn=predict_next_questions, inputs="number", outputs="number")
21
+
22
  intf.launch()
23
 
24