Add query similar movies to "Shawshank Redemption, The (1994)"

Author: Daniel-Robbins

examples/chDB_vector_search.ipynb

@@ -670,6 +670,80 @@
         "print(\"Movie Id, Title, Embeddings\")\n",
         "print(chs.query('SELECT * FROM embeddings_with_title LIMIT 5'))\n"
       ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "target_movieId = 318\n",
+        "topN = chs.query(f\"\"\"\n",
+        "          WITH\n",
+        "            {target_movieId} AS theMovieId,\n",
+        "            (SELECT embedding FROM embeddings_with_title WHERE movieId = theMovieId LIMIT 1) AS targetEmbedding\n",
+        "          SELECT\n",
+        "            movieId,\n",
+        "            title,\n",
+        "            cosineDistance(embedding, targetEmbedding) AS distance\n",
+        "            FROM embeddings_with_title\n",
+        "            WHERE movieId != theMovieId -- Not self\n",
+        "            ORDER BY distance ASC\n",
+        "            LIMIT 10\n",
+        "          \"\"\", \"Pretty\")\n",
+        "print(f\"Scaned {topN.rows_read()} rows, \"\n",
+        "      f\"Top 10 similar movies to movieId {target_movieId} in {topN.elapsed()}\")\n",
+        "print(\"Target Movie:\")\n",
+        "print(chs.query(f\"SELECT * FROM movielens.movies WHERE movieId={target_movieId}\", \"Pretty\"))\n",
+        "print(\"Top10 Similar:\")\n",
+        "print(topN)\n"
+      ],
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "wvBnbe4Vn7O4",
+        "outputId": "825530f5-012f-438a-bb33-13ecec4ac542"
+      },
+      "execution_count": 37,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Scaned 10 rows, Top 10 similar movies to movieId 318 in 0.037762344\n",
+            "Target Movie:\n",
+            "┏━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━┓\n",
+            "┃ \u001b[1mmovieId\u001b[0m ┃ \u001b[1mtitle                           \u001b[0m ┃ \u001b[1mgenres     \u001b[0m ┃\n",
+            "┡━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━┩\n",
+            "│     318 │ Shawshank Redemption, The (1994) │ Crime|Drama │\n",
+            "└─────────┴──────────────────────────────────┴─────────────┘\n",
+            "\n",
+            "Top10 Similar:\n",
+            "┏━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━┓\n",
+            "┃ \u001b[1mmovieId\u001b[0m ┃ \u001b[1mtitle                           \u001b[0m ┃ \u001b[1m   distance\u001b[0m ┃\n",
+            "┡━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━┩\n",
+            "│     527 │ Schindler's List (1993)          │ 0.045725048 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│     593 │ Silence of the Lambs, The (1991) │ 0.069214344 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│      50 │ Usual Suspects, The (1995)       │  0.10683352 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│     296 │ Pulp Fiction (1994)              │  0.11023432 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│     356 │ Forrest Gump (1994)              │  0.15409416 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│     858 │ Godfather, The (1972)            │  0.16235054 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│     110 │ Braveheart (1995)                │  0.17049736 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│    2028 │ Saving Private Ryan (1998)       │   0.1742953 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│    2858 │ American Beauty (1999)           │  0.19775331 │\n",
+            "├─────────┼──────────────────────────────────┼─────────────┤\n",
+            "│    2959 │ Fight Club (1999)                │  0.19789273 │\n",
+            "└─────────┴──────────────────────────────────┴─────────────┘\n",
+            "\n"
+          ]
+        }
+      ]
     }
   ],
   "metadata": {