scope_rl.ope.weight_value_learning.function.StateWeightFunction#

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

State Weight Function (for both discrete and continuous action space).

Bases: torch.nn.Module

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

Parameters:

  • 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