From 955654323921e3a12a894356a8b0883035ecb0f2 Mon Sep 17 00:00:00 2001 From: Tirumalesh <111861663+tiruk007@users.noreply.github.com> Date: Thu, 6 Oct 2022 15:47:00 +0530 Subject: [PATCH] Fix for ranking example Fix for ranking example, Fixes issue #57921 Attached working [gist] (https://colab.research.google.com/gist/tiruk007/6e46d215661dea14a142ca82231b5505/landingpage_example.ipynb#scrollTo=n3oYt3R6Nr9l) for reference. --- tensorflow_ranking/examples/keras/landingpage_example.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow_ranking/examples/keras/landingpage_example.ipynb b/tensorflow_ranking/examples/keras/landingpage_example.ipynb index bf91964..9843ff8 100644 --- a/tensorflow_ranking/examples/keras/landingpage_example.ipynb +++ b/tensorflow_ranking/examples/keras/landingpage_example.ipynb @@ -43,7 +43,7 @@ " \u003ca target=\"_blank\" href=\"https://www.tensorflow.org/ranking/overview\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/tf_logo_32px.png\" /\u003eView on TensorFlow.org\u003c/a\u003e\n", " \u003c/td\u003e\n", " \u003ctd\u003e\n", - " \u003ca target=\"_blank\" href=\"https://github.com/tensorflow/ranking/blob/master/examples/keras/landingpage_example.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/GitHub-Mark-32px.png\" /\u003eView source on GitHub\u003c/a\u003e\n", + " \u003ca target=\"_blank\" href=\"https://github.com/tensorflow/ranking/blob/master/tensorflow_ranking/examples/keras/landingpage_example.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/GitHub-Mark-32px.png\" /\u003eView source on GitHub\u003c/a\u003e\n", " \u003c/td\u003e\n", "\u003c/table\u003e" ] @@ -101,7 +101,7 @@ "\n", "# Create a model with Keras Functional API.\n", "inputs = {\n", - " name: tf.keras.Input(shape=(None, 1), dtype=tf.float32, name=name)\n", + " name: tf.keras.Input(shape=(None, 136), dtype=tf.float32, name=name)\n", " for name in ds.element_spec[0]\n", " if name != \"_mask\"\n", "}\n",