trinity.algorithm.entropy_loss_fn.entropy_loss_fn module

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.MixEntropyLossFn(entropy_coef: float)[source]#

Bases: EntropyLossFn

Basic entropy loss function for mix algorithm.

__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]#
Contents