Skip to content

Commit cacacc8

Browse files
committed
FIX meta-learning
1 parent 65a50cd commit cacacc8

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

autosklearn/metalearning/metalearning/kNearestDatasets/kND.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def fit(self, metafeatures, runs):
5454
best_configuration_per_dataset[dataset_name] = None
5555
else:
5656
best_configuration_per_dataset[dataset_name] = \
57-
runs[dataset_name].argmin(skipna=True)
57+
runs[dataset_name].index[np.nanargmin(runs[dataset_name].values)]
5858

5959
self.best_configuration_per_dataset = best_configuration_per_dataset
6060

autosklearn/metalearning/metalearning/meta_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def __init__(self, configuration_space, aslib_directory):
4444
for algorithm_id in self.configurations:
4545
configuration = self.configurations[algorithm_id]
4646
try:
47-
configurations[algorithm_id] = \
47+
configurations[str(algorithm_id)] = \
4848
(Configuration(configuration_space, values=configuration))
4949
except (ValueError, KeyError) as e:
5050
self.logger.debug("Error reading configurations: %s", e)
@@ -97,7 +97,7 @@ def _get_metafeatures(self, features):
9797
return all_metafeatures
9898

9999
def get_configuration_from_algorithm_index(self, idx):
100-
return self.configurations[idx]
100+
return self.configurations[str(idx)]
101101
#configuration = self.configurations[idx]
102102
#configuration = Configuration(self.configuration_space,
103103
# **configuration)

0 commit comments

Comments
 (0)