SNN2.src.model.reinforcement.reinforcement module

class SNN2.src.model.reinforcement.reinforcement.RLModelHandler(pp: ~SNN2.src.decorators.decorators.c_logger.<locals>.augmented_cls, rewardFunction: ~SNN2.src.decorators.decorators.c_logger.<locals>.augmented_cls, RLPerfEvaluation: ~SNN2.src.decorators.decorators.c_logger.<locals>.augmented_cls, ObsPP: ~SNN2.src.decorators.decorators.c_logger.<locals>.augmented_cls, ActPolicy: ~SNN2.src.decorators.decorators.c_logger.<locals>.augmented_cls, *args, **kwargs)

Bases: DefaultModelHandler

calculate_reward(*args, **kwargs) → int

execute_episode(*args, **kwargs) → int

get_action(*args, **kwargs) → int

get_correct_wrong(obtained: Tensor, expected: Tensor) → Tuple[Tensor, Tensor]

log_history_state() → None

log_memory_state() → None