mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType > Class Template Reference

Forward declaration of NStepQLearningWorker. More...

#include <async_learning.hpp>

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

Public Member Functions
	NStepQLearningWorker (const UpdaterType &updater, const EnvironmentType &environment, const TrainingConfig &config, bool deterministic)
	Construct N-step Q-Learning worker with the given parameters and environment. More...
	NStepQLearningWorker (const NStepQLearningWorker &other)
	Copy another NStepQLearningWorker. More...
	NStepQLearningWorker (NStepQLearningWorker &&other)
	Take ownership of another NStepQLearningWorker. More...
NStepQLearningWorker &	operator= (const NStepQLearningWorker &other)
	Copy another NStepQLearningWorker. More...
NStepQLearningWorker &	operator= (NStepQLearningWorker &&other)
	Take ownership of another NStepQLearningWorker. More...
	~NStepQLearningWorker ()
	Clean memory.
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::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >

Forward declaration of NStepQLearningWorker.

N-step Q-Learning 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.

Constructor & Destructor Documentation

NStepQLearningWorker() [1/3]

template<typename EnvironmentType , typename NetworkType , typename UpdaterType , typename PolicyType >

mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >::NStepQLearningWorker ( const UpdaterType &  updater, const EnvironmentType &  environment, const TrainingConfig &  config, bool  deterministic  )

inline

Construct N-step Q-Learning 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.

NStepQLearningWorker() [2/3]

template<typename EnvironmentType , typename NetworkType , typename UpdaterType , typename PolicyType >

mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >::NStepQLearningWorker ( const NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType > &  other )

inline

Copy another NStepQLearningWorker.

Parameters

other

NStepQLearningWorker to copy.

NStepQLearningWorker() [3/3]

template<typename EnvironmentType , typename NetworkType , typename UpdaterType , typename PolicyType >

mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >::NStepQLearningWorker ( NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType > &&  other )

inline

Take ownership of another NStepQLearningWorker.

Parameters

other

NStepQLearningWorker to take ownership of.

Member Function Documentation

Initialize()

template<typename EnvironmentType , typename NetworkType , typename UpdaterType , typename PolicyType >

void mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >::Initialize ( NetworkType &  learningNetwork )

inline

Initialize the worker.

Parameters

learningNetwork

The shared network.

operator=() [1/2]

template<typename EnvironmentType , typename NetworkType , typename UpdaterType , typename PolicyType >

NStepQLearningWorker& mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >::operator= ( const NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType > &  other )

inline

Copy another NStepQLearningWorker.

Parameters

other

NStepQLearningWorker to copy.

operator=() [2/2]

template<typename EnvironmentType , typename NetworkType , typename UpdaterType , typename PolicyType >

NStepQLearningWorker& mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >::operator= ( NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType > &&  other )

inline

Take ownership of another NStepQLearningWorker.

Parameters

other

NStepQLearningWorker to take ownership of.

Step()

template<typename EnvironmentType , typename NetworkType , typename UpdaterType , typename PolicyType >

bool mlpack::rl::NStepQLearningWorker< EnvironmentType, NetworkType, UpdaterType, PolicyType >::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.

---

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

• src/mlpack/methods/reinforcement_learning/async_learning.hpp
• src/mlpack/methods/reinforcement_learning/worker/n_step_q_learning_worker.hpp