5
5
class BiasedSVD (BiasedSVDScorer ):
6
6
def __init__ (self , features , ** kwargs ):
7
7
super ().__init__ (features = features , ** kwargs )
8
+ self .features = features # Store features as an instance variable for testing
8
9
9
10
@staticmethod
10
11
def get_default_configspace (number_item : int , ** kwargs ):
@@ -22,8 +23,8 @@ def get_default_configspace(number_item: int, **kwargs):
22
23
features = UniformIntegerHyperparameter ('features' , lower = 2 , upper = n_items , default_value = n_items - 1 , log = True )
23
24
else :
24
25
#features = Integer('features', bounds=(2, 10000), default=1000, log=True) # No default values given
25
- features = UniformIntegerHyperparameter ('features' , lower = 2 , upper = 10000 , log = True )
26
- damping = UniformFloatHyperparameter ('damping' , lower = 0.0001 , upper = 1000 , log = True )
26
+ features = UniformIntegerHyperparameter ('features' , lower = 2 , upper = 10000 , default_value = 1000 , log = True )
27
+ damping = UniformFloatHyperparameter ('damping' , lower = 0.0001 , upper = 1000 , default_value = 5 , log = True )
27
28
28
29
cs = ConfigurationSpace ()
29
30
cs .add ([features , damping ])
0 commit comments