arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove a warning when computing left_proba

Browse Source
This commit is contained in:
Aurélien Geron
2024-01-19 19:01:59 +13:00
parent b2a2533543
commit 4f5aa43f92
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
18_reinforcement_learning.ipynb
View File

@@ -807,7 +807,7 @@
"# extra code – a function that creates an animation for a given policy model\n",
"\n",
"def pg_policy(obs):\n",
" left_proba = model.predict(obs[np.newaxis], verbose=0)\n",
" left_proba = model.predict(obs[np.newaxis], verbose=0)[0][0]\n",
" return int(np.random.rand() > left_proba)\n",
"\n",
"np.random.seed(42)\n",