示例#1
0
def test_l2_proximal_operator(beta, expected):
    npt.assert_array_equal(regs.L2().proximal_operator(beta, 1), expected)
示例#2
0
def test_l2_hessian(beta):
    npt.assert_array_equal(regs.L2().hessian(beta), np.eye(len(beta)))
示例#3
0
def test_l2_gradient(beta):
    npt.assert_array_equal(regs.L2().gradient(beta), beta)
示例#4
0
def test_l2_function(beta, expected):
    assert regs.L2().f(beta) == expected
示例#5
0
def test_elastic_net_gradient_zero_weight_is_l2():
    beta = np.array([1, 2, 3])
    npt.assert_array_equal(
        regs.ElasticNet(weight=0).gradient(beta),
        regs.L2().gradient(beta))
示例#6
0
def test_elastic_net_function_zero_weight_is_l2():
    beta = np.array([1, 2, 3])
    assert regs.ElasticNet(weight=0).f(beta) == regs.L2().f(beta)