RLConfigComponentFramework

class srl.rl.models.config.framework_config.RLConfigComponentFramework(framework: Literal['auto', 'tensorflow', 'torch'] = 'auto')

ベースクラス: object

set_tensorflow()

use tensorflow

set_torch()

use torch

set_auto()

use tensorflow/torch インストールされている方を採用します。 両方インストールされている場合はTensorflowが優先されます。

InputBlock

class srl.rl.models.config.input_block.InputBlockConfig(value: srl.rl.models.config.input_block.InputValueBlockConfig = <factory>, image: srl.rl.models.config.input_block.InputImageBlockConfig = <factory>)

ベースクラス: object

InputValueBlock

class srl.rl.models.config.input_block.InputValueBlockConfig(name: str = '', kwargs: dict = <factory>, processors: List[srl.base.rl.processor.RLProcessor] = <factory>)

ベースクラス: object

InputImageBlock

class srl.rl.models.config.input_block.InputImageBlockConfig(name: str = '', kwargs: dict = <factory>, processors: Optional[List[srl.base.rl.processor.RLProcessor]] = None)

ベースクラス: object

set_dqn_block(filters: int = 32, activation: str = 'relu')

画像の入力に対してDQNで採用されたLayersを使用します。

パラメータ:

  • filters (int, optional) -- 基準となるfilterの数です.

  • activation (str) -- activation function. Defaults to "relu".

set_r2d3_block(filters: int = 16, activation: str = 'relu')

画像の入力に対してR2D3で採用されたLayersを使用します。

パラメータ:

  • filters (int, optional) -- 基準となるfilterの数です.

  • activation (str, optional) -- activation function. Defaults to "relu".

set_alphazero_block(n_blocks: int = 19, filters: int = 256, activation: str = 'relu')

Alphaシリーズの画像レイヤーで使用する層を指定します。 AlphaZeroで採用されている層です。

パラメータ:

  • n_blocks (int, optional) -- ブロック数. Defaults to 19.

  • filters (int, optional) -- フィルター数. Defaults to 256.

  • activation (str, optional) -- activation function. Defaults to "relu".

set_muzero_atari_block(filters: int = 128, activation: str = 'relu', use_layer_normalization: bool = False)

Alphaシリーズの画像レイヤーで使用する層を指定します。 MuZeroのAtari環境で採用されている層です。

パラメータ:

  • filters (int, optional) -- フィルター数. Defaults to 128.

  • activation (str, optional) -- activation function. Defaults to "relu".

  • use_layer_normalization (str, optional) -- use_layer_normalization. Defaults to True.

HiddenBlock

class srl.rl.models.config.hidden_block.HiddenBlockConfig(name: str = '', kwargs: dict = <factory>)

ベースクラス: object

set(layer_sizes: Tuple[int, ...], activation: str = 'relu', **kwargs)

Multi-layer Perceptron Block

パラメータ:

  • layer_sizes (Tuple[int, ...], optional) -- 各レイヤーのユニット数. Defaults to (512,).

  • activation (str, optional) -- Activation function. Defaults to "relu".

サンプル

>>> conf = HiddenBlockConfig()
>>> conf.set((128, 64, 32))

DuelingNetwork

class srl.rl.models.config.dueling_network.DuelingNetworkConfig(name: str = '', kwargs: dict = <factory>)

ベースクラス: object

set(layer_sizes: Tuple[int, ...], activation: str = 'relu', **kwargs)

Multi-layer Perceptron Block

パラメータ:

  • layer_sizes (Tuple[int, ...], optional) -- 各レイヤーのユニット数. Defaults to (512,).

  • activation (str, optional) -- Activation function. Defaults to "relu".

サンプル

>>> conf = DuelingNetworkConfig()
>>> conf.set((128, 64, 32))
set_dueling_network(layer_sizes: Tuple[int, ...], activation: str = 'relu', dueling_type: Literal['', 'average', 'max'] = 'average', **kwargs)

Multi-layer Perceptron Block + DuelingNetwork Block

パラメータ:

  • layer_sizes (Tuple[int, ...], optional) -- 各レイヤーのユニット数. Defaults to (512,).

  • activation (str, optional) -- Activation function. Defaults to "relu".

  • dueling_type (str, optional) -- select algorithm. Defaults to "average".