Skip to content

Commit 81f905b

Browse files
committed
made the parameter feature accessible for testing, added the default value for damping + Test for svd is now OK
1 parent cd9c5fd commit 81f905b

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

lkauto/algorithms/svd.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
class BiasedSVD(BiasedSVDScorer):
66
def __init__(self, features, **kwargs):
77
super().__init__(features=features, **kwargs)
8+
self.features = features # Store features as an instance variable for testing
89

910
@staticmethod
1011
def get_default_configspace(number_item: int, **kwargs):
@@ -22,8 +23,8 @@ def get_default_configspace(number_item: int, **kwargs):
2223
features = UniformIntegerHyperparameter('features', lower=2, upper=n_items, default_value=n_items-1, log=True)
2324
else:
2425
#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)
2728

2829
cs = ConfigurationSpace()
2930
cs.add([features, damping])

tests/algorithms/test_svd.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,16 +7,16 @@
77

88
class TestSVD(unittest.TestCase):
99
def test_init_givenObjectInitialized_ObjectInitializedCorrectlyExpected(self):
10-
features = 10
11-
svd = BiasedSVD(features)
12-
self.assertIsInstance(svd, BiasedSVD)
13-
self.assertEqual(features, svd.factorization.n_components)
10+
expected_features = 10
11+
biased_svd = BiasedSVD(expected_features)
12+
self.assertIsInstance(biased_svd, BiasedSVD)
13+
self.assertEqual(expected_features, biased_svd.features)
1414

1515
def test_defaultConfigspace_GivenFunctionCalled_correctConfigSpaceReturnedExpected(self):
1616
number_item_list = [1500, 100000]
1717
for number_item in number_item_list:
1818
cs = BiasedSVD.get_default_configspace(number_item=number_item)
19-
params = cs.get_hyperparameters()
19+
params = list(cs.values())
2020
with self.subTest(number_item=number_item):
2121
self.assertIsInstance(cs, CS.ConfigurationSpace)
2222
if number_item < 10000:

0 commit comments

Comments
 (0)