@@ -44,13 +44,14 @@ class LinearLearner(AmazonAlgorithmEstimatorBase):
4444 init_sigma = hp ('init_sigma' , gt (0 ), 'A float greater-than 0' , float )
4545 init_bias = hp ('init_bias' , (), 'A number' , float )
4646 optimizer = hp ('optimizer' , isin ('sgd' , 'adam' , 'auto' ), 'One of "sgd", "adam" or "auto' , str )
47- loss = hp ('loss' , isin ('logistic' , 'squared_loss' , 'absolute_loss' , 'auto' ),
47+ loss = hp ('loss' , isin ('logistic' , 'squared_loss' , 'absolute_loss' , 'hinge_loss' , 'eps_insensitive_squared_loss' ,
48+ 'eps_insensitive_absolute_loss' , 'quantile_loss' , 'huber_loss' , 'auto' ),
4849 '"logistic", "squared_loss", "absolute_loss", "hinge_loss", "eps_insensitive_squared_loss", '
4950 '"eps_insensitive_absolute_loss", "quantile_loss", "huber_loss" or "auto"' , str )
5051 wd = hp ('wd' , ge (0 ), 'A float greater-than or equal to 0' , float )
5152 l1 = hp ('l1' , ge (0 ), 'A float greater-than or equal to 0' , float )
5253 momentum = hp ('momentum' , (ge (0 ), lt (1 )), 'A float in [0,1)' , float )
53- learning_rate = hp ('learning_rate' , gt (0 ), 'A float greater-than or equal to 0' , float )
54+ learning_rate = hp ('learning_rate' , gt (0 ), 'A float greater-than 0' , float )
5455 beta_1 = hp ('beta_1' , (ge (0 ), lt (1 )), 'A float in [0,1)' , float )
5556 beta_2 = hp ('beta_2' , (ge (0 ), lt (1 )), 'A float in [0,1)' , float )
5657 bias_lr_mult = hp ('bias_lr_mult' , gt (0 ), 'A float greater-than 0' , float )
0 commit comments