Skip to content

Commit 5f3d273

Browse files
authored
Update preprocessing_layers.ipynb
bug fix ValueError: Multi-dimensional indexing (e.g. `obj[:, None]`) is no longer supported. Convert to a numpy array before indexing instead.
1 parent 60abeb0 commit 5f3d273

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

site/en/tutorials/structured_data/preprocessing_layers.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -297,7 +297,7 @@
297297
"def df_to_dataset(dataframe, shuffle=True, batch_size=32):\n",
298298
" df = dataframe.copy()\n",
299299
" labels = df.pop('target')\n",
300-
" df = {key: value[:,tf.newaxis] for key, value in dataframe.items()}\n",
300+
" df = {key: value.to_numpy()[:,tf.newaxis] for key, value in dataframe.items()}\n",
301301
" ds = tf.data.Dataset.from_tensor_slices((dict(df), labels))\n",
302302
" if shuffle:\n",
303303
" ds = ds.shuffle(buffer_size=len(dataframe))\n",

0 commit comments

Comments
 (0)