schuetoliver
/
handson-ml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #396 from lebaste77/master 

very minor change on greedy policy variable usage
main
Aurélien Geron 2021-03-02 10:16:43 +13:00 committed by GitHub
parent b201196be1 64f0e05a94
commit 47538082f8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
18_reinforcement_learning.ipynb
View File

@ -1306,7 +1306,7 @@
"source": [ "source": [
"def epsilon_greedy_policy(state, epsilon=0):\n", "def epsilon_greedy_policy(state, epsilon=0):\n",
" if np.random.rand() < epsilon:\n", " if np.random.rand() < epsilon:\n",
" return np.random.randint(2)\n", " return np.random.randint(n_outputs)\n",
" else:\n", " else:\n",
" Q_values = model.predict(state[np.newaxis])\n", " Q_values = model.predict(state[np.newaxis])\n",
" return np.argmax(Q_values[0])" " return np.argmax(Q_values[0])"