We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1bdebe9 commit 65a50cdCopy full SHA for 65a50cd
autosklearn/data/abstract_data_manager.py
@@ -63,10 +63,13 @@ def perform1HotEncoding(self):
63
64
self.info['is_sparse'] = 1 if sparse else 0
65
self.data['X_train'] = data[0]
66
- if 'X_valid' in self.data:
+ if 'X_valid' in self.data and 'X_test' in self.data:
67
self.data['X_valid'] = data[1]
68
- if 'X_test' in self.data:
69
self.data['X_test'] = data[2]
+ elif 'X_valid' in self.data:
70
+ self.data['X_valid'] = data[1]
71
+ elif 'X_test' in self.data:
72
+ self.data['X_test'] = data[1]
73
74
def __repr__(self):
75
return 'DataManager : ' + self.name
0 commit comments