Machine Learning

Please see Graph algorithms for an introduction to the maturity tiers: production-quality, beta and alpha.

1. Pipeline Catalog

1.1. Beta Tier

Table 1. List of all beta pipeline catalog operations in the GDS library.
Description Operation

Check if a pipeline exists

gds.beta.pipeline.exists

Remove a pipeline from memory

gds.beta.pipeline.drop

List pipelines

gds.beta.pipeline.list

2. Model Catalog

2.1. Beta Tier

Table 2. List of all beta model catalog operations in the GDS library. Functions are written in italic.
Description Operation

Check if a model exists

gds.beta.model.exists

Remove a model from memory

gds.beta.model.drop

List models

gds.beta.model.list

2.2. Alpha Tier

Table 3. List of all alpha model catalog operations in the GDS library. Functions are written in italic.
Description Operation

Store a model

gds.alpha.model.store

Load a stored model

gds.alpha.model.load

Delete a stored model

gds.alpha.model.delete

Publish a model

gds.alpha.model.publish

3. Pipelines

3.1. Beta tier

Table 4. List of all beta machine learning pipelines operations in the GDS library. Functions are written in italic.
Algorithm name Operation

Link Prediction Pipeline

gds.beta.pipeline.linkPrediction.create

gds.beta.pipeline.linkPrediction.addNodeProperty

gds.beta.pipeline.linkPrediction.addFeature

gds.beta.pipeline.linkPrediction.addLogisticRegression

gds.beta.pipeline.linkPrediction.addRandomForest

gds.beta.pipeline.linkPrediction.configureSplit

gds.beta.pipeline.linkPrediction.train

gds.beta.pipeline.linkPrediction.train.estimate

gds.beta.pipeline.linkPrediction.predict.mutate

gds.beta.pipeline.linkPrediction.predict.mutate.estimate

gds.beta.pipeline.linkPrediction.predict.stream

gds.beta.pipeline.linkPrediction.predict.stream.estimate

Node Classification Pipeline

gds.beta.pipeline.nodeClassification.create

gds.beta.pipeline.nodeClassification.addNodeProperty

gds.beta.pipeline.nodeClassification.selectFeatures

gds.beta.pipeline.nodeClassification.addLogisticRegression

gds.beta.pipeline.nodeClassification.addRandomForest

gds.beta.pipeline.nodeClassification.configureSplit

gds.beta.pipeline.nodeClassification.train

gds.beta.pipeline.nodeClassification.train.estimate

gds.beta.pipeline.nodeClassification.predict.mutate

gds.beta.pipeline.nodeClassification.predict.mutate.estimate

gds.beta.pipeline.nodeClassification.predict.stream

gds.beta.pipeline.nodeClassification.predict.stream.estimate

gds.beta.pipeline.nodeClassification.predict.write

gds.beta.pipeline.nodeClassification.predict.write.estimate

3.2. Alpha tier

Table 5. List of all alpha machine learning pipelines operations in the GDS library. Functions are written in italic.
Algorithm name Operation

Link Prediction Pipeline

gds.alpha.pipeline.linkPrediction.addMLP

gds.alpha.pipeline.linkPrediction.configureAutoTuning

gds.alpha.pipeline.linkPrediction.addRandomForest (deprecated)

Node Classification Pipeline

gds.alpha.pipeline.nodeClassification.addMLP

gds.alpha.pipeline.nodeClassification.configureAutoTuning

gds.alpha.pipeline.nodeClassification.addRandomForest (deprecated)

Node Regression Pipeline

gds.alpha.pipeline.nodeRegression.create

gds.alpha.pipeline.nodeRegression.addNodeProperty

gds.alpha.pipeline.nodeRegression.selectFeatures

gds.alpha.pipeline.nodeRegression.configureAutoTuning

gds.alpha.pipeline.nodeRegression.configureSplit

gds.alpha.pipeline.nodeRegression.addLinearRegression

gds.alpha.pipeline.nodeRegression.addRandomForest

gds.alpha.pipeline.nodeRegression.train

gds.alpha.pipeline.nodeRegression.predict.stream

gds.alpha.pipeline.nodeRegression.predict.mutate

4. Node embeddings

4.1. Production-quality tier

Table 6. List of all production-quality node embedding algorithms in the GDS library. Functions are written in italic.
Algorithm name Operation

Fast Random Projection

gds.fastRP.mutate

gds.fastRP.mutate.estimate

gds.fastRP.stats

gds.fastRP.stats.estimate

gds.fastRP.stream

gds.fastRP.stream.estimate

gds.fastRP.write

gds.fastRP.write.estimate

4.2. Beta tier

Table 7. List of all beta node embedding algorithms in the GDS library. Functions are written in italic.
Algorithm name Operation

GraphSAGE

gds.beta.graphSage.stream

gds.beta.graphSage.stream.estimate

gds.beta.graphSage.mutate

gds.beta.graphSage.mutate.estimate

gds.beta.graphSage.write

gds.beta.graphSage.write.estimate

gds.beta.graphSage.train

gds.beta.graphSage.train.estimate

HashGNN

gds.beta.hashgnn.mutate

gds.beta.hashgnn.mutate.estimate

gds.beta.hashgnn.stream

gds.beta.hashgnn.stream.estimate

Node2Vec

gds.beta.node2vec.mutate

gds.beta.node2vec.mutate.estimate

gds.beta.node2vec.stream

gds.beta.node2vec.stream.estimate

gds.beta.node2vec.write

gds.beta.node2vec.write.estimate