Struct GridWorldEnvironment 

pub struct GridWorldEnvironment { /* private fields */ }

GridWorld environment for testing reinforcement learning

Implementations

impl GridWorldEnvironment

pub fn new(width: usize, height: usize) -> Self

Creates a new GridWorld environment

pub fn with_goal(self, x: usize, y: usize) -> Self

Sets the goal position

pub fn with_obstacles(self, obstacles: Vec<(usize, usize)>) -> Self

Sets the obstacles

pub fn is_obstacle(&self, x: usize, y: usize) -> bool

Checks if a position is an obstacle

pub fn is_goal(&self, x: usize, y: usize) -> bool

Checks if a position is the goal

Trait Implementations

impl Environment for GridWorldEnvironment

fn state(&self) -> Array1<f64>

Gets the current state

fn num_actions(&self) -> usize

Gets the number of available actions

fn step(&mut self, action: usize) -> Result<(Array1<f64>, f64, bool)>

Takes an action and returns the reward and next state

fn reset(&mut self) -> Array1<f64>

Resets the environment