update (examples): adjusted to the new query approach

examples/README.md

@@ -20,135 +20,70 @@ SELECT * FROM disease_embedding LIMIT 3;
 * Similarity: 0.9657052160858545
 
 ~~~sql
-WITH
-d1 AS (
-    SELECT embedding
-    FROM disease d, disease_embedding de
-    WHERE de.disease_id = d.disease_id
-          AND d.name = 'colon cancer'
-),
-d2 AS (
-    SELECT embedding
-    FROM disease d, disease_embedding de
-    WHERE de.disease_id = d.disease_id
-          AND d.name = 'liver cancer'
-)
-
 SELECT
-    list_sum(
-        list_transform(
-            list_zip(d1.embedding, d2.embedding),
-            x -> x[1] * x[2]
-        )
-    )
-    /
-    (
-        sqrt(list_sum(list_transform(d1.embedding, x -> x * x)))
-        *
-        sqrt(list_sum(list_transform(d2.embedding, x -> x * x)))
-    )
-AS cosine_similarity
-FROM d1, d2;
+    cosine(e1.embedding, e2.embedding) AS similarity
+FROM disease_embedding e1, disease d1, disease_embedding e2, disease d2
+WHERE d1.disease_id = e1.disease_id
+      AND d2.disease_id = e2.disease_id
+      AND d1.name = 'colon cancer'
+      AND d2.name = 'liver cancer';
 ~~~
 
 ### Similarity between `cirrhosis of the liver` and `liver cancer`:
 * Similarity: 0.9498555648670207
 
 ~~~sql
-WITH
-d1 AS (
-    SELECT embedding
-    FROM disease d, disease_embedding de
-    WHERE de.disease_id = d.disease_id
-          AND d.name = 'cirrhosis of the liver'
-),
-d2 AS (
-    SELECT embedding
-    FROM disease d, disease_embedding de
-    WHERE de.disease_id = d.disease_id
-          AND d.name = 'liver cancer'
-)
-
 SELECT
-    list_sum(
-        list_transform(
-            list_zip(d1.embedding, d2.embedding),
-            x -> x[1] * x[2]
-        )
-    )
-    /
-    (
-        sqrt(list_sum(list_transform(d1.embedding, x -> x * x)))
-        *
-        sqrt(list_sum(list_transform(d2.embedding, x -> x * x)))
-    )
-AS cosine_similarity
-FROM d1, d2;
+    cosine(e1.embedding, e2.embedding) AS similarity
+FROM disease_embedding e1, disease d1, disease_embedding e2, disease d2
+WHERE d1.disease_id = e1.disease_id
+      AND d2.disease_id = e2.disease_id
+      AND d1.name = 'cirrhosis of the liver'
+      AND d2.name = 'liver cancer';
 ~~~
 
 ### Similarity between `common cold` and `liver cancer`:
 * Similarity: 0.7145078607670947
 
 ~~~sql
-WITH
-d1 AS (
-    SELECT embedding
-    FROM disease d, disease_embedding de
-    WHERE de.disease_id = d.disease_id
-          AND d.name = 'common cold'
-),
-d2 AS (
-    SELECT embedding
-    FROM disease d, disease_embedding de
-    WHERE de.disease_id = d.disease_id
-          AND d.name = 'liver cancer'
-)
+SELECT
+    cosine(e1.embedding, e2.embedding) AS similarity
+FROM disease_embedding e1, disease d1, disease_embedding e2, disease d2
+WHERE d1.disease_id = e1.disease_id
+      AND d2.disease_id = e2.disease_id
+      AND d1.name = 'common cold'
+      AND d2.name = 'liver cancer';
+~~~
+
+### Rank diseases by symptoms `fever` and `cough`:
 
+~~~sql
 SELECT
-    list_sum(
-        list_transform(
-            list_zip(d1.embedding, d2.embedding),
-            x -> x[1] * x[2]
-        )
-    )
-    /
-    (
-        sqrt(list_sum(list_transform(d1.embedding, x -> x * x)))
-        *
-        sqrt(list_sum(list_transform(d2.embedding, x -> x * x)))
-    )
-AS cosine_similarity
-FROM d1, d2;
+    d.name,
+    cosine(
+        embed_symptoms(['fever','cough']),
+        de.embedding
+    ) AS similarity
+FROM disease d, disease_embedding de
+WHERE d.disease_id = de.disease_id
+ORDER BY similarity DESC;
 ~~~
 
 ### Rank Diseases by Similarity to `common cold`:
 
 ~~~sql
-WITH target AS (
-    SELECT embedding
-    FROM disease d, disease_embedding de
-    WHERE de.disease_id = d.disease_id
-          AND d.name = 'common cold'
-)
-
 SELECT
     d.name,
-    (
-        list_sum(
-            list_transform(
-                list_zip(target.embedding, de.embedding),
-                x -> x[1] * x[2]
-            )
-        )
-        /
+    cosine(
         (
-            sqrt(list_sum(list_transform(target.embedding, x -> x * x)))
-            *
-            sqrt(list_sum(list_transform(de.embedding, x -> x * x)))
-        )
+            SELECT de.embedding
+            FROM disease_embedding de
+            JOIN disease d2 ON d2.disease_id = de.disease_id
+            WHERE d2.name = 'common cold'
+        ),
+        de.embedding
     ) AS similarity
-FROM disease_embedding de
-JOIN disease d ON de.disease_id = d.disease_id,
-target
+FROM disease d, disease_embedding de
+WHERE d.disease_id = de.disease_id
 ORDER BY similarity DESC;
 ~~~