thatgeeman
commited on
Commit
•
03ec795
1
Parent(s):
6ea878a
Update README.md
Browse files
README.md
CHANGED
@@ -25,6 +25,26 @@ model-index:
|
|
25 |
This is a trained model of a **PPO** agent playing **LunarLander-v2**
|
26 |
using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
|
27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
28 |
## Usage (with Stable-baselines3)
|
29 |
TODO: Add your code
|
30 |
|
|
|
25 |
This is a trained model of a **PPO** agent playing **LunarLander-v2**
|
26 |
using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
|
27 |
|
28 |
+
## Model parameters
|
29 |
+
```python
|
30 |
+
model = PPO(
|
31 |
+
policy = 'MlpPolicy',
|
32 |
+
env = env,
|
33 |
+
n_steps = 1024,
|
34 |
+
batch_size = 64,
|
35 |
+
n_epochs = 10,
|
36 |
+
gamma = 0.999,
|
37 |
+
gae_lambda = 0.98,
|
38 |
+
ent_coef = 0.01,
|
39 |
+
verbose=1)
|
40 |
+
|
41 |
+
```
|
42 |
+
Trained for 10^6 steps using
|
43 |
+
|
44 |
+
```python
|
45 |
+
steps = 1e6
|
46 |
+
model.learn(total_timesteps=int(steps))
|
47 |
+
```
|
48 |
## Usage (with Stable-baselines3)
|
49 |
TODO: Add your code
|
50 |
|