Struct RogueNetAgent

pub struct RogueNetAgent { /* private fields */ }

Agent that implements the RogueNet entity neural network. Can be loaded from checkpoints produced by enn-trainer.

Implementations

impl RogueNetAgent

pub fn load<P: AsRef<Path>>(path: P) -> Result<Self, Error>

Loads a neural network agent from an enn-trainer checkpoint directory.

pub fn load_archive<R: Read>(reader: R) -> Result<Self, Error>

Loads a neural network agent from an archive of a checkpoint directory.

You can use the rogue-net cli to create a tar archive of a checkpoint directory:

$ cargo install rogue-net
$ rogue-net archive --path path/to/checkpoint/dir

pub fn with_feature_adaptor<E: Featurizable>(self) -> Self

Adapts the network to a changed observation space.

If you trained a network with a different observation space, you can adapt it to the new observation space. For this to work, the new set of features of the observation space must be a superset of the old set of features.

Trait Implementations

impl Agent for RogueNetAgent

fn act_dyn( &mut self, _action: &str, _num_actions: u64, obs: &Obs, ) -> Option<Vec<u64>>

Returns an action for the given observation.

fn act_async_dyn( &mut self, action: &str, num_actions: u64, obs: &Obs, ) -> ActionReceiver<u64>

Returns receiver that can be blocked on to receive an action for the given observation.

fn game_over(&mut self, _: &Obs)

Indicates that the agent has reached the end of the training episode.

impl Clone for RogueNetAgent

fn clone(&self) -> RogueNetAgent

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.