def test_ftrl_set_individual(): ft = Ftrl(double_precision=tparams.double_precision) ft.alpha = tparams.alpha ft.beta = tparams.beta ft.lambda1 = tparams.lambda1 ft.lambda2 = tparams.lambda2 ft.nbins = tparams.nbins ft.nepochs = tparams.nepochs assert ft.params == tparams
def test_ftrl_set_individual(): ft = Ftrl() ft.alpha = tparams.alpha ft.beta = tparams.beta ft.lambda1 = tparams.lambda1 ft.lambda2 = tparams.lambda2 ft.nbins = tparams.nbins ft.nepochs = tparams.nepochs ft.interactions = tparams.interactions assert ft.params == tparams
def test_ftrl_set_individual(): ft = Ftrl(double_precision=tparams.double_precision) ft.alpha = tparams.alpha ft.beta = tparams.beta ft.lambda1 = tparams.lambda1 ft.lambda2 = tparams.lambda2 ft.nbins = tparams.nbins ft.mantissa_nbits = tparams.mantissa_nbits ft.nepochs = tparams.nepochs ft.negative_class = tparams.negative_class ft.interactions = tparams.interactions assert ft.params == tparams
def test_ftrl_set_bad_beta_value(value): ft = Ftrl() with pytest.raises(ValueError) as e: ft.beta = value assert ("Value should be greater than or equal to zero: %s" % str(value) == str(e.value))
def test_ftrl_set_wrong_beta_type(): ft = Ftrl() with pytest.raises(TypeError) as e: ft.beta = "-1.0" assert ("Expected a float, instead got <class 'str'>" == str(e.value))