gfn.envs.preprocessors.hot

Module Contents

Classes

KHotPreprocessor	Base class for Preprocessors. The goal is to transform tensors representing raw states
OneHotPreprocessor	Base class for Preprocessors. The goal is to transform tensors representing raw states

Attributes

OutputTensor

class gfn.envs.preprocessors.hot.KHotPreprocessor(height, ndim, get_states_indices)

Bases: gfn.envs.preprocessors.base.Preprocessor

Base class for Preprocessors. The goal is to transform tensors representing raw states to tensors that can be used as input to neural networks.

Parameters

• height (int) –

• ndim (int) –

• get_states_indices (Callable[[gfn.containers.states.States], OutputTensor]) –

name = k_hot

preprocess(states)

class gfn.envs.preprocessors.hot.OneHotPreprocessor(n_states, get_states_indices)

Bases: gfn.envs.preprocessors.base.Preprocessor

Base class for Preprocessors. The goal is to transform tensors representing raw states to tensors that can be used as input to neural networks.

Parameters

• n_states (int) –

• get_states_indices (Callable[[gfn.containers.states.States], OutputTensor]) –

name = one_hot

preprocess(states)

gfn.envs.preprocessors.hot.OutputTensor