reagent.net_builder.slate_reward package

Submodules

reagent.net_builder.slate_reward.slate_reward_gru module

class reagent.net_builder.slate_reward.slate_reward_gru.SlateRewardGRU(gru: reagent.core.parameters.GRUParameters = <factory>, fit_slate_wise_reward: bool = True)

Bases: reagent.net_builder.slate_reward_net_builder.SlateRewardNetBuilder

build_slate_reward_network(state_dim, candidate_dim, candidate_size, slate_size) → reagent.models.base.ModelBase

property expect_slate_wise_reward

fit_slate_wise_reward: bool = True

gru: reagent.core.parameters.GRUParameters

reagent.net_builder.slate_reward.slate_reward_transformer module

class reagent.net_builder.slate_reward.slate_reward_transformer.SlateRewardTransformer(transformer: reagent.core.parameters.TransformerParameters = <factory>, fit_slate_wise_reward: bool = True)

Bases: reagent.net_builder.slate_reward_net_builder.SlateRewardNetBuilder

build_slate_reward_network(state_dim, candidate_dim, candidate_size, slate_size) → reagent.models.base.ModelBase

property expect_slate_wise_reward

fit_slate_wise_reward: bool = True

transformer: reagent.core.parameters.TransformerParameters

Module contents

class reagent.net_builder.slate_reward.SlateRewardNetBuilder__Union(SlateRewardGRU: Optional[reagent.net_builder.slate_reward.slate_reward_gru.SlateRewardGRU] = None, SlateRewardTransformer: Optional[reagent.net_builder.slate_reward.slate_reward_transformer.SlateRewardTransformer] = None)

Bases: reagent.core.tagged_union.TaggedUnion

SlateRewardGRU: Optional[reagent.net_builder.slate_reward.slate_reward_gru.SlateRewardGRU] = None

SlateRewardTransformer: Optional[reagent.net_builder.slate_reward.slate_reward_transformer.SlateRewardTransformer] = None