Python linear_behavior_policy_logit 예제들

예제 #1

파일: test_synthetic_slate_functions.py 프로젝트: aiueola/zr-obp

def test_linear_behavior_policy_logit_using_invalid_input(
        context, action_context, tau, err, description):
    if description == "":
        with pytest.raises(err):
            linear_behavior_policy_logit(context=context,
                                         action_context=action_context,
                                         tau=tau)
    else:
        with pytest.raises(err, match=f"{description}*"):
            linear_behavior_policy_logit(context=context,
                                         action_context=action_context,
                                         tau=tau)

예제 #2

def test_linear_behavior_policy_logit_using_valid_input(
    context, action_context, tau, description
):
    logit_value = linear_behavior_policy_logit(
        context=context, action_context=action_context, tau=tau
    )
    assert logit_value.shape == (context.shape[0], action_context.shape[0])