how to run more episodes?

[SHITIANYU-hue]

in this script: https://colab.research.google.com/github/eleurent/highway-env/blob/master/scripts/highway_planning.ipynb

The code only runs one episode, how could I run more episodes to update the agent, and save the model?

[eleurent]

This agent is a planning algorithm (Monte-Carlo Tree Search), not a reinforcement learning algorithm : it samples transitions from a known dynamics to find an optimal trajectory starting from the current state. Thus, there is nothing to update and save.

It could be used in a model-based RL fashion, where a model of the dynamics is learned from observed transitions, and then this model is used for planning, as in this script. But it is quite a different goal, and it's not implemented at the moment.