This commit is contained in:
Michael Pilosov 2022-04-01 19:52:11 -06:00
parent c7613d0bca
commit f85551a16e
2 changed files with 2 additions and 2 deletions

View File

@ -33,7 +33,7 @@ Therefore, since our objective is to stabilize the cart, the target "time series
You can generate your own data with: You can generate your own data with:
```bash ```bash
python data.py python sample.py
``` ```
Note: if you change the presumed sample space in `data.py`, you should make the corresponding changes to the initial distribution in `main.py`. Note: if you change the presumed sample space in `data.py`, you should make the corresponding changes to the initial distribution in `main.py`.

View File

@ -20,7 +20,7 @@ def train(data):
D["u"] = D["i"] * D["o"] / D["p"] D["u"] = D["i"] * D["o"] / D["p"]
mud_point_idx = D["u"].argmax() mud_point_idx = D["u"].argmax()
mud_point = D["lam"].iloc[mud_point_idx] mud_point = D["lam"].iloc[mud_point_idx]
print(f"MUD Point ({mud_point_idx}: {mud_point}") print(f"MUD Point {mud_point_idx}: {mud_point}")
return mud_point return mud_point