CS_ALearn/deep-rl-class
1
0
Fork 0
mirror of https://github.com/huggingface/deep-rl-class.git synced 2026-04-03 10:38:27 +08:00
Code Issues Projects Releases Wiki Activity

unit2: Use tqdm in train template

Browse Source
This commit is contained in:
Johannes 'fish' Ziemke
2022-12-14 17:22:42 +01:00
committed by GitHub
parent e6e6b1f9af
commit d4657698be
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
notebooks/unit2/unit2.ipynb
View File

@@ -771,7 +771,7 @@
"outputs": [],
"source": [
"def train(n_training_episodes, min_epsilon, max_epsilon, decay_rate, env, max_steps, Qtable):\n",
" for episode in range(n_training_episodes):\n",
" for episode in tqdm(range(n_training_episodes))\n",
" # Reduce epsilon (because we need less and less exploration)\n",
" epsilon = min_epsilon + (max_epsilon - min_epsilon)*np.exp(-decay_rate*episode)\n",
" # Reset the environment\n",
@@ -1748,4 +1748,4 @@
},
"nbformat": 4,
"nbformat_minor": 0
}
}