Merge pull request #1 from orazve/kge-predict-add-write-notebook-api

Add write mode demonstration

Author: orazve

examples/kge-predict-transe-pyg-train.ipynb

@@ -577,6 +577,57 @@
     "print(result)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "collapsed": false
+   },
+   "source": [
+    "## Using Write Mode\n",
+    "\n",
+    "Write mode allows you to write results directly to the database as a new relationship type. This approach helps to avoid mapping from `nodeId` to `id`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "write_relationship_type = \"PREDICTED_\" + rel_label_to_predict\n",
+    "result_write = transe_model.predict_write(\n",
+    "    source_node_filter=source_ids_df.nodeId,\n",
+    "    target_node_filter=\"Entity\",\n",
+    "    relationship_type=rel_label_to_predict,\n",
+    "    write_relationship_type=write_relationship_type,\n",
+    "    write_property=\"transe_score\",\n",
+    "    top_k=3,\n",
+    "    concurrency=4,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "collapsed": false
+   },
+   "source": [
+    "Extract the result from the database."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gds.run_cypher(\n",
+    "    \"MATCH (n)-[r:\"\n",
+    "    + write_relationship_type\n",
+    "    + \"]->(m) RETURN n.id AS sourceId, n.text AS sourceTag, m.id AS targetId, m.text AS targetTag, r.transe_score AS score\"\n",
+    ")"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,