Skip to content

Commit de56d42

Browse files
committed
Merge branch 'hmendozap-hotfix_SparseInput' into development_java
2 parents 8ae3c67 + f4e67ef commit de56d42

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

autosklearn/data/xy_data_manager.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,12 @@ def __init__(self, data_x, y, task, metric, feat_type, dataset_name,
2323

2424
self.info['task'] = task
2525
self.info['metric'] = metric
26-
self.info['is_sparse'] = 1 if sparse.issparse(data_x) else 0
27-
self.info['has_missing'] = np.all(np.isfinite(data_x))
26+
if sparse.issparse(data_x):
27+
self.info['is_sparse'] = 1
28+
self.info['has_missing'] = np.all(np.isfinite(data_x.data))
29+
else:
30+
self.info['is_sparse'] = 0
31+
self.info['has_missing'] = np.all(np.isfinite(data_x))
2832

2933
label_num = {
3034
REGRESSION: 1,

0 commit comments

Comments
 (0)