RLConfigComponentFramework

class srl.rl.models.config.framework_config.RLConfigComponentFramework(framework: str = 'auto')

ベースクラス: object

set_tensorflow(): use tensorflow

set_torch(): use torch

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

RLConfigComponentInput

class srl.rl.models.config.input_config.RLConfigComponentInput

ベースクラス: object

input_value_block: MLPBlockConfig: <MLPBlock>

input_image_block: InputImageBlockConfig: <InputImageBlock>

InputImageBlock

class srl.rl.models.config.input_config.InputImageBlockConfig(_name: str = '', _kwargs: dict = <factory>, _processors: List[srl.base.rl.processor.RLProcessor] = <factory>)

ベースクラス: object

set_dqn_block(image_type: SpaceTypes = SpaceTypes.GRAY_2ch, resize: Tuple[int, int] = (84, 84), filters: int = 32, activation: str = 'relu')

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

パラメータ:

image_type (SpaceTypes) -- 画像のタイプ. Defaults to SpaceTypes.GRAY_2ch
resize (Tuple[int, int]) -- 画像のサイズ. Defaults to (84, 84)
filters (int) -- 基準となるfilterの数です. Defaults to 32.
activation (str) -- activation function. Defaults to "relu".

set_r2d3_block(image_type: SpaceTypes = SpaceTypes.COLOR, resize: Tuple[int, int] = (96, 72), filters: int = 16, activation: str = 'relu')

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

パラメータ:

image_type (SpaceTypes) -- 画像のタイプ. Defaults to SpaceTypes.COLOR
resize (Tuple[int, int]) -- 画像のサイズ. Defaults to (96, 72)
filters (int, optional) -- 基準となるfilterの数です. Defaults to 32.
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(image_type: SpaceTypes = SpaceTypes.GRAY_2ch, resize: Tuple[int, int] = (96, 96), 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.

MLPBlock

class srl.rl.models.config.mlp_block.MLPBlockConfig

ベースクラス: object

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

Multi-layer Perceptron Block

パラメータ:

layer_sizes (Tuple[int, ...], optional) -- 各レイヤーのユニット数. Defaults to (512,).
activation (str, optional) -- Activation function. Defaults to "relu".

サンプル

>>> mlp_conf = MLPBlockConfig()
>>> mlp_conf.set((128, 64, 32))

DuelingNetwork

class srl.rl.models.config.dueling_network.DuelingNetworkConfig

ベースクラス: object

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

Multi-layer Perceptron Block

パラメータ:

layer_sizes (Tuple[int, ...], optional) -- 各レイヤーのユニット数. Defaults to (512,).
activation (str, optional) -- Activation function. Defaults to "relu".

サンプル

>>> mlp_conf = DuelingNetworkConfig()
>>> mlp_conf.set((128, 64, 32))

set_dueling_network(layer_sizes: Tuple[int, ...] = (512,), activation: str = 'relu', dueling_type: str = '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".