Skip to content

Commit 9ebca02

Browse files
Update sklbench/datasets/loaders.py
Co-authored-by: david-cortes-intel <david.cortes@intel.com>
1 parent 164be60 commit 9ebca02

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sklbench/datasets/loaders.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -717,8 +717,8 @@ def load_szilard_10m(
717717
X_test_raw = d_test.drop(columns=[label_col])
718718

719719
combined = pd.concat([X_train_raw, X_test_raw], axis=0, ignore_index=True)
720-
X_combined_oh = pd.get_dummies(combined)
721-
x = sparse.csr_matrix(X_combined_oh.values)
720+
X_combined_oh = pd.get_dummies(combined, sparse=True)
721+
x = sparse.csr_matrix(X_combined_oh)
722722

723723
n_train = len(d_train)
724724
n_test = len(d_test)

0 commit comments

Comments
 (0)