scope_rl.ope.weight_value_learning.function.ContinuousStateActionWeightFunction#

class scope_rl.ope.weight_value_learning.function.ContinuousStateActionWeightFunction(action_dim, state_dim, hidden_dim=100, enable_gradient_reversal=False)[source]#

State Action Weight Function (for continuous action space).

Bases: torch.nn.Module

Imported as: scope_rl.ope.weight_value_learning.function.ContinuousStateActionWeightFunction

Parameters:

  • action_dim (int (> 0)) – Dimensions of the action space.

  • state_dim (int (> 0)) – Dimensions of the state space.

  • hidden_dim (int, default=100 (> 0)) – Hidden dimension of the network.

  • enable_gradient_reversal (bool = False) – Whether to enable gradient reversal layer (for loss maximization).

Methods