d3rlpy.algos.GreedyTransformerActionSampler¶
- class d3rlpy.algos.GreedyTransformerActionSampler(*args, **kwargs)[source]¶
Greedy action-sampler.
This class implements greedy function to determine action from discrte probability distribution.
Methods
- __call__(transformer_output)[source]¶
Returns sampled action from Transformer output.
- Parameters
transformer_output (numpy.ndarray[Any, numpy.dtype[Any]]) – Output of Transformer algorithms.
- Returns
Sampled action.
- Return type
Union[numpy.ndarray[Any, numpy.dtype[Any]], int]