Skip to content

Commit 64397cb

Browse files
committed
MAINT fix bug related to subsampling
1 parent 151dc0c commit 64397cb

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

autosklearn/evaluation/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,6 @@ def run(self, config, instance=None,
104104
if instance_specific is None or instance_specific == '0':
105105
instance_specific = {}
106106
else:
107-
print(instance_specific)
108107
instance_specific = [specific.split('=') for specific in instance_specific.split(',')]
109108
instance_specific = {specific[0]: specific[1] for specific in instance_specific}
110109
subsample = instance_specific.get('subsample')

autosklearn/evaluation/train_evaluator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,8 +230,9 @@ def subsample_indices(self, train_indices):
230230
stratify = None
231231

232232
if len(train_indices) > self.subsample:
233+
indices = np.arange(len(train_indices))
233234
cv_indices_train, _ = sklearn.cross_validation.train_test_split(
234-
train_indices, stratify=stratify,
235+
indices, stratify=stratify,
235236
train_size=self.subsample, random_state=1)
236237
train_indices = train_indices[cv_indices_train]
237238
return train_indices

0 commit comments

Comments
 (0)