Skip to content

Commit b252075

Browse files
Fixed cfproto adult OHE example. Replaced OHE sparse with OHE dense and fixed display function.
1 parent e95e66b commit b252075

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

doc/source/examples/cfproto_cat_adult_ohe.ipynb

+5-8
Original file line numberDiff line numberDiff line change
@@ -265,9 +265,7 @@
265265
"X_num = X[:, -4:].astype(np.float32, copy=False)\n",
266266
"xmin, xmax = X_num.min(axis=0), X_num.max(axis=0)\n",
267267
"rng = (-1., 1.)\n",
268-
"X_num_scaled = (X_num - xmin) / (xmax - xmin) * (rng[1] - rng[0]) + rng[0]\n",
269-
"X_num_scaled_train = X_num_scaled[:idx, :]\n",
270-
"X_num_scaled_test = X_num_scaled[idx+1:, :]"
268+
"X_num_scaled = (X_num - xmin) / (xmax - xmin) * (rng[1] - rng[0]) + rng[0]"
271269
]
272270
},
273271
{
@@ -284,8 +282,7 @@
284282
"outputs": [],
285283
"source": [
286284
"X_cat = X[:, :-4].copy()\n",
287-
"ohe = OneHotEncoder(categories='auto')\n",
288-
"ohe.fit(X_cat)\n",
285+
"ohe = OneHotEncoder(categories='auto', sparse=False).fit(X_cat)\n",
289286
"X_cat_ohe = ohe.transform(X_cat)"
290287
]
291288
},
@@ -310,7 +307,7 @@
310307
}
311308
],
312309
"source": [
313-
"X = np.c_[X_cat_ohe.todense(), X_num_scaled].astype(np.float32, copy=False)\n",
310+
"X = np.c_[X_cat_ohe, X_num_scaled].astype(np.float32, copy=False)\n",
314311
"X_train, X_test = X[:idx, :], X[idx+1:, :]\n",
315312
"print(X_train.shape, X_test.shape)"
316313
]
@@ -601,8 +598,8 @@
601598
" print('\\nNumerical:')\n",
602599
" delta_num = X_cf_ord[0, -4:] - X_orig_ord[0, -4:]\n",
603600
" n_keys = len(list(cat_vars_ord.keys()))\n",
604-
" for i in range(delta_num.shape[1]):\n",
605-
" if np.abs(delta_num[0, i]) > eps:\n",
601+
" for i in range(delta_num.shape[0]):\n",
602+
" if np.abs(delta_num[i]) > eps:\n",
606603
" print('{}: {:.2f} --> {:.2f}'.format(feature_names[i+n_keys],\n",
607604
" X_orig_ord[0,i+n_keys],\n",
608605
" X_cf_ord[0,i+n_keys]))"

0 commit comments

Comments
 (0)