Merge pull request #536 from memgraph/gnn_link_prediction

[master<]Gnn link prediction
memgraph · Oct 31, 2022 · 32cffb7 · 32cffb7
2 parents 50fecb2 + 2bfe7cf
commit 32cffb7
Show file tree

Hide file tree

Showing 6 changed files with 379 additions and 3 deletions.
diff --git a/...odules/python/gnn-link-prediction/gnn-link-prediction-example-visualization.png b/...odules/python/gnn-link-prediction/gnn-link-prediction-example-visualization.png
diff --git a/...odules/python/gnn-link-prediction/gnn-link-prediction-neighborhood-sampling.png b/...odules/python/gnn-link-prediction/gnn-link-prediction-neighborhood-sampling.png
diff --git a/...s/python/gnn-link-prediction/gnn-link-prediction-transductive-dataset-split.png b/...s/python/gnn-link-prediction/gnn-link-prediction-transductive-dataset-split.png
diff --git a/mage/query-modules/python/gnn-link-prediction.md b/mage/query-modules/python/gnn-link-prediction.md
diff --git a/mage/templates/_mage_spells.mdx b/mage/templates/_mage_spells.mdx
@@ -17,6 +17,7 @@
 | [import_util](/mage/query-modules/python/import-util)                                  | Python | A module for importing data from different formats (JSON).                                                                                                                                                                        |
 | [katz_centrality](/mage/query-modules/cpp/katz-centrality)                             | C++    | Katz centrality is a centrality measurement that outputs a node's influence based on the number of shortest paths and their weighted length.                                                                                      |
 | [katz_centrality_online](/mage/query-modules/cpp/katz-centrality-online)               | C++    | Online implementation of the Katz centrality. Outputs the approximate result for Katz centrality while maintaining the order of rankings.                                                                                         |
+| [link_prediction_with_gnn](/mage/query-modules/python/link-prediction-with-gnn)             | Python | Module for predicting links in the graph by using graph neural networks.                                                                                                                                                          |
 | [max_flow](/mage/query-modules/python/max-flow)                                        | Python | An algorithm for finding a flow through a graph such that it is the maximum possible flow.                                                                                                                                        |
 | [node2vec](/mage/query-modules/python/node2vec)                                        | Python | An algorithm for calculating node embeddings on static graph.                                                                                                                                                                     |
 | [node2vec_online](/mage/query-modules/python/node2vec-online)                          | Python | An algorithm for calculating node embeddings as new edges arrive.                                                                                                                                                                 |

diff --git a/sidebars/sidebarsMAGE.js b/sidebars/sidebarsMAGE.js
@@ -4,7 +4,7 @@ module.exports = {
     {
       type: "category",
       label: "Install MAGE",
-      link: {type: 'doc', id: 'installation/overview'},
+      link: { type: 'doc', id: 'installation/overview' },
       items: [
         "installation/docker-hub",
         "installation/docker-build",
@@ -20,7 +20,7 @@ module.exports = {
     {
       type: "category",
       label: "Available query modules",
-      link: {type: 'doc', id: 'query-modules/available-queries'},
+      link: { type: 'doc', id: 'query-modules/available-queries' },
       items: [
         "query-modules/cpp/betweenness-centrality",
         "query-modules/cpp/betweenness-centrality-online",
@@ -39,6 +39,7 @@ module.exports = {
         "query-modules/python/json-util",
         "query-modules/cpp/katz-centrality",
         "query-modules/cpp/katz-centrality-online",
+        "query-modules/python/link-prediction-with-gnn",
         "query-modules/python/max-flow",
         "query-modules/python/node2vec",
         "query-modules/python/node2vec-online",
@@ -67,7 +68,7 @@ module.exports = {
     {
       type: "category",
       label: "Algorithms",
-      link: {type: 'doc', id: 'algorithms/overview'},
+      link: { type: 'doc', id: 'algorithms/overview' },
       items: [
         {
           type: "category",