Skip to content

Commit

Permalink
update to compatible dtype no more dtype manipulation allowed in work…
Browse files Browse the repository at this point in the history
…flow (#1503)

Co-authored-by: Ben Frederickson <[email protected]>
  • Loading branch information
jperez999 and benfred authored Apr 6, 2022
1 parent aae71a3 commit e151b01
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions examples/getting-started-movielens/03-Training-with-TF.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -427,7 +427,7 @@
"outputs": [],
"source": [
"for col in CATEGORICAL_COLUMNS:\n",
" inputs[col] = tf.keras.Input(name=col, dtype=tf.int32, shape=(1,))\n",
" inputs[col] = tf.keras.Input(name=col, dtype=tf.int64, shape=(1,))\n",
"# Note that we need two input tensors for multi-hot categorical features\n",
"for col in CATEGORICAL_MH_COLUMNS:\n",
" inputs[col] = (tf.keras.Input(name=f\"{col}__values\", dtype=tf.int64, shape=(1,)),\n",
Expand Down Expand Up @@ -674,10 +674,7 @@
"metadata": {},
"outputs": [],
"source": [
"workflow = nvt.Workflow.load(os.path.join(INPUT_DATA_DIR, \"workflow\"))\n",
"\n",
"workflow.output_dtypes[\"userId\"] = \"int32\"\n",
"workflow.output_dtypes[\"movieId\"] = \"int32\""
"workflow = nvt.Workflow.load(os.path.join(INPUT_DATA_DIR, \"workflow\"))"
]
},
{
Expand Down Expand Up @@ -737,7 +734,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.12"
"version": "3.8.10"
}
},
"nbformat": 4,
Expand Down

0 comments on commit e151b01

Please sign in to comment.