OneStepSarsaWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType > Class Template Reference

Forward declaration of OneStepSarsaWorker. More...

Public Types
using	ActionType = typename EnvironmentType::Action
using	StateType = typename EnvironmentType::State
using	TransitionType = std::tuple< StateType, ActionType, double, StateType, ActionType >

Public Member Functions
	OneStepSarsaWorker (const UpdaterType &updater, const EnvironmentType &environment, const TrainingConfig &config, bool deterministic)
	Construct one step sarsa worker with the given parameters and environment. More...
void	Initialize (NetworkType &learningNetwork)
	Initialize the worker. More...
bool	Step (NetworkType &learningNetwork, NetworkType &targetNetwork, size_t &totalSteps, PolicyType &policy, double &totalReward)
	The agent will execute one step. More...

Detailed Description

template

<

typename

EnvironmentType

,

typename

NetworkType

,

typename

UpdaterType

,

typename

PolicyType

>

class mlpack::rl::OneStepSarsaWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >

Forward declaration of OneStepSarsaWorker.

One step Sarsa worker.

Template Parameters

EnvironmentType	The type of the reinforcement learning task.
NetworkType	The type of the network model.
UpdaterType	The type of the optimizer.
PolicyType	The type of the behavior policy.

Definition at line 163 of file async_learning.hpp.

Member Typedef Documentation

ActionType

using ActionType = typename EnvironmentType::Action

Definition at line 39 of file one_step_sarsa_worker.hpp.

StateType

using StateType = typename EnvironmentType::State

Definition at line 38 of file one_step_sarsa_worker.hpp.

TransitionType

using TransitionType = std::tuple<StateType, ActionType, double, StateType, ActionType>

Definition at line 41 of file one_step_sarsa_worker.hpp.

Constructor & Destructor Documentation

OneStepSarsaWorker()

OneStepSarsaWorker ( const UpdaterType &  updater, const EnvironmentType &  environment, const TrainingConfig &  config, bool  deterministic  )

inline

Construct one step sarsa worker with the given parameters and environment.

Parameters

updater	The optimizer.
environment	The reinforcement learning task.
config	Hyper-parameters.
deterministic	Whether it should be deterministic.

Definition at line 52 of file one_step_sarsa_worker.hpp.

Member Function Documentation

Initialize()

void Initialize ( NetworkType &  learningNetwork )

inline

Initialize the worker.

Parameters

learningNetwork

The shared network.

Definition at line 68 of file one_step_sarsa_worker.hpp.

Step()

bool Step ( NetworkType &  learningNetwork, NetworkType &  targetNetwork, size_t &  totalSteps, PolicyType &  policy, double &  totalReward  )

inline

The agent will execute one step.

Parameters

learningNetwork	The shared learning network.
targetNetwork	The shared target network.
totalSteps	The shared counter for total steps.
policy	The shared behavior policy.
totalReward	This will be the episode return if the episode ends after this step. Otherwise this is invalid.

Returns

Indicate whether current episode ends after this step.

Definition at line 87 of file one_step_sarsa_worker.hpp.

References TrainingConfig::Discount(), TrainingConfig::GradientLimit(), TrainingConfig::StepLimit(), TrainingConfig::StepSize(), TrainingConfig::TargetNetworkSyncInterval(), and TrainingConfig::UpdateInterval().

---

The documentation for this class was generated from the following files:

• /home/ryan/src/mlpack.org/_src/mlpack-3.1.0/src/mlpack/methods/reinforcement_learning/async_learning.hpp
• /home/ryan/src/mlpack.org/_src/mlpack-3.1.0/src/mlpack/methods/reinforcement_learning/worker/one_step_sarsa_worker.hpp