Expand description

The ai module provides a reinforcement learning agent that controls a train in a railway network simulation.

The module defines the TrainAgentState struct, which represents the state of a train agent in the simulation, and the TrainAgentAction enum, which represents the possible actions a train agent can take in the simulation.

The main component of this module is the TrainAgentRL struct, which implements the rurel::mdp::Agent trait for the train agent. This struct handles the reinforcement learning process, updating the train agent’s state based on the actions taken and the rewards received.

Structs

  • A reinforcement learning agent that controls a train in the simulation.
  • Reinforcement Learning Agent for controlling a train in the simulation.
  • Represents the state of a train agent in the simulation.