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))
