"text/plain": "array([b\"'Til There Was You (1997)\", b'1-900 (1994)',\n b'101 Dalmatians (1996)', b'12 Angry Men (1957)', b'187 (1997)',\n b'2 Days in the Valley (1996)',\n b'20,000 Leagues Under the Sea (1954)',\n b'2001: A Space Odyssey (1968)',\n b'3 Ninjas: High Noon At Mega Mountain (1998)',\n b'39 Steps, The (1935)'], dtype=object)"
"WARNING:tensorflow:From /home/mao/anaconda3/envs/tf2.5/lib/python3.6/site-packages/tensorflow/python/ops/array_ops.py:5049: calling gather (from tensorflow.python.ops.array_ops) with validate_indices is deprecated and will be removed in a future version.\n",
"Instructions for updating:\n",
"The `validate_indices` argument has no effect. Indices are always validated on CPU and never validated on GPU.\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"WARNING:tensorflow:From /home/mao/anaconda3/envs/tf2.5/lib/python3.6/site-packages/tensorflow/python/ops/array_ops.py:5049: calling gather (from tensorflow.python.ops.array_ops) with validate_indices is deprecated and will be removed in a future version.\n",
"Instructions for updating:\n",
"The `validate_indices` argument has no effect. Indices are always validated on CPU and never validated on GPU.\n"
"print(f\"Recommendations for user 42: {titles[0, :3]}\")"
],
"metadata": {
"collapsed": false,
"pycharm": {
"name": "#%%\n"
}
}
},
{
"cell_type": "code",
"execution_count": 21,
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"WARNING:absl:Found untraced functions such as query_with_exclusions while saving (showing 1 of 1). These functions will not be directly callable after loading.\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"INFO:tensorflow:Assets written to: /tmp/tmp2gxi6mvw/model/assets\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"INFO:tensorflow:Assets written to: /tmp/tmp2gxi6mvw/model/assets\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"WARNING:tensorflow:No training configuration found in save file, so the model was *not* compiled. Compile it manually.\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"WARNING:tensorflow:No training configuration found in save file, so the model was *not* compiled. Compile it manually.\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"Recommendations: [b'Rudy (1993)' b'Homeward Bound: The Incredible Journey (1993)'\n",
" b'Client, The (1994)']\n"
]
}
],
"source": [
"# Export the query model.\n",
"with tempfile.TemporaryDirectory() as tmp:\n",
" path = os.path.join(tmp, \"model\")\n",
"\n",
" # Save the index.\n",
" index.save(path)\n",
"\n",
" # Load it back; can also be done in TensorFlow Serving.\n",
" loaded = tf.keras.models.load_model(path)\n",
"\n",
" # Pass a user id in, get top predicted movie titles back.\n",
" scores, titles = loaded([\"42\"])\n",
"\n",
" print(f\"Recommendations: {titles[0][:3]}\")"
],
"metadata": {
"collapsed": false,
"pycharm": {
"name": "#%%\n"
}
}
},
{
"cell_type": "code",
"execution_count": 22,
"outputs": [
{
"data": {
"text/plain": "<tensorflow_recommenders.layers.factorized_top_k.ScaNN at 0x7ff528665630>"
"print(f\"Recommendations for user 42: {titles[0, :3]}\")"
],
"metadata": {
"collapsed": false,
"pycharm": {
"name": "#%%\n"
}
}
},
{
"cell_type": "code",
"execution_count": 24,
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"WARNING:absl:Found untraced functions such as query_with_exclusions while saving (showing 1 of 1). These functions will not be directly callable after loading.\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"INFO:tensorflow:Assets written to: /tmp/tmpg2n407yk/model/assets\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"INFO:tensorflow:Assets written to: /tmp/tmpg2n407yk/model/assets\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"WARNING:tensorflow:No training configuration found in save file, so the model was *not* compiled. Compile it manually.\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"WARNING:tensorflow:No training configuration found in save file, so the model was *not* compiled. Compile it manually.\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"Recommendations: [b'Rudy (1993)' b'Homeward Bound: The Incredible Journey (1993)'\n",