agent.py
def critic_1(
            self,
            critic_params: hk.Params,
            state_action: np.ndarray
    ) -> jnp.DeviceArray:
        """Retrieves the result from a single critic network. Relevant for the actor update rule."""
        return self.critic.apply(critic_params, state_action)[0].squeeze(-1)
Retrieves the result from a single critic network. Relevant for the actor update rule.
Retrieves the result from a single critic network. Relevant for the actor update rule.
Retrieves the result from a single critic network . Relevant for the actor update rule .
def critic_1(
            self,
            critic_params: hk.Params,
            state_action: np.ndarray
    ) -> jnp.DeviceArray:
        return self.critic.apply(critic_params, state_action)[0].squeeze(-1)
def critic_1 ( self , critic_params : hk . Params , state_action : np . ndarray ) -> jnp . DeviceArray : return self . critic . apply ( critic_params , state_action ) [ 0 ] . squeeze ( - 1 )
Retrieves the result from a single critic network.
Retrieves the result from a single critic network .
"""Retrieves the result from a single critic network. Relevant for the actor update rule."""
self
critic_params
hk.Params
state_action
np.ndarray