Skip to content

Commit 6350fef

Browse files
Remove use of deprecated start_index from Categorify (#1134)
Co-authored-by: rnyak <16246900+rnyak@users.noreply.github.com>
1 parent 94eda09 commit 6350fef

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

examples/usecases/ecommerce-session-based-next-item-prediction-for-fashion.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -468,7 +468,7 @@
468468
"source": [
469469
"%%time\n",
470470
"item_features_names = ['f_' + str(col) for col in [47, 68]]\n",
471-
"cat_features = [['item_id', 'purchase_id']] + item_features_names >> nvt.ops.Categorify(start_index=1)\n",
471+
"cat_features = [['item_id', 'purchase_id']] + item_features_names >> nvt.ops.Categorify()\n",
472472
"\n",
473473
"features = ['session_id', 'timestamp'] + cat_features"
474474
]

examples/usecases/transformers-next-item-prediction.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,7 @@
352352
"\n",
353353
"categorical_features = (['city_id', 'booker_country', 'hotel_country'] +\n",
354354
" weekday_checkin + weekday_checkout\n",
355-
" ) >> ops.Categorify(start_index=1) \n",
355+
" ) >> ops.Categorify()\n",
356356
"\n",
357357
"groupby_features = categorical_features + ['utrip_id', 'checkin'] >> ops.Groupby(\n",
358358
" groupby_cols=['utrip_id'],\n",

merlin/datasets/ecommerce/booking/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ def default_booking_transformation(**kwargs):
183183
Returns:
184184
Workflow: NVTabular workflow
185185
"""
186-
cat = lambda: nvt.ops.Categorify(start_index=1) # noqa: E731
186+
cat = lambda: nvt.ops.Categorify() # noqa: E731
187187

188188
df_season = get_lib().DataFrame(
189189
{"month": range(1, 13), "season": ([0] * 3) + ([1] * 3) + ([2] * 3) + ([3] * 3)}

0 commit comments

Comments
 (0)