lit_sb3.modules.feature_extractors.base

source module lit_sb3.modules.feature_extractors.base

Classes

• BaseFeaturesExtractor — Base class that represents a features extractor.

source class BaseFeaturesExtractor(observation_space: gym.Space, features_dim: int = 0)

Bases : nn.Module

Base class that represents a features extractor.

Parameters

• observation_space : gym.Space — The observation space of the environment

• features_dim : int — Number of features extracted.

Attributes

• features_dim : int — The number of features that the extractor outputs.

source property BaseFeaturesExtractor.features_dim: int

The number of features that the extractor outputs.