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 23d4e6f commit 29a023cCopy full SHA for 29a023c
1 file changed
openml/datasets/dataset.py
@@ -531,8 +531,11 @@ def _encode_if_category(column):
531
'PyOpenML cannot handle string when returning numpy'
532
' arrays. Use dataset_format="dataframe".'
533
)
534
- elif array_format == "dataframe" and scipy.sparse.issparse(data):
535
- return pd.SparseDataFrame(data, columns=attribute_names)
+ elif array_format == "dataframe":
+ if scipy.sparse.issparse(data):
536
+ return pd.SparseDataFrame(data, columns=attribute_names)
537
+ else:
538
+ return data
539
else:
540
data_type = "sparse-data" if scipy.sparse.issparse(data) else "non-sparse data"
541
logging.warning(
0 commit comments