trinity.algorithm.entropy_loss_fn

Submodules

trinity.algorithm.entropy_loss_fn.entropy_loss_fn module

class trinity.algorithm.entropy_loss_fn.entropy_loss_fn.EntropyLossFn[source]

Bases: ABC

Entropy loss function.

classmethod default_args() → Dict[source]
Returns:

The default arguments for the entropy loss function.

Return type:

Dict

class trinity.algorithm.entropy_loss_fn.entropy_loss_fn.DefaultEntropyLossFn(entropy_coef: float)[source]

Bases: EntropyLossFn

Basic entropy loss function.

__init__(entropy_coef: float)[source]
class trinity.algorithm.entropy_loss_fn.entropy_loss_fn.DummyEntropyLossFn(entropy_coef: float)[source]

Bases: EntropyLossFn

Dummy entropy loss function.

__init__(entropy_coef: float)[source]

Module contents

class trinity.algorithm.entropy_loss_fn.EntropyLossFn[source]

Bases: ABC

Entropy loss function.

classmethod default_args() → Dict[source]
Returns:

The default arguments for the entropy loss function.

Return type:

Dict