EntropyTuner

rlox_nn::traits

Trait EntropyTuner 

Source 
pub trait EntropyTuner {
    // Required methods
    fn alpha(&self) -> f32;
    fn update(
        &mut self,
        log_probs: &TensorData,
        target_entropy: f32,
    ) -> Result<f64, NNError>;
}
Expand description

Entropy tuning for SAC.

Required Methods§

Source

fn alpha(&self) -> f32

Source

fn update( &mut self, log_probs: &TensorData, target_entropy: f32, ) -> Result<f64, NNError>

Implementors§