oracle.pgx.api.mllib

Class UnsupervisedGraphWiseModel

java.lang.Object

oracle.pgx.api.mllib.Model<ModelType>

oracle.pgx.api.mllib.GraphWiseModel<UnsupervisedGraphWiseModelConfig,oracle.pgx.api.internal.mllib.UnsupervisedGraphWiseModelMetadata,UnsupervisedGraphWiseModel>

oracle.pgx.api.mllib.UnsupervisedGraphWiseModel

All Implemented Interfaces:

java.lang.AutoCloseable

public class UnsupervisedGraphWiseModel
extends GraphWiseModel<UnsupervisedGraphWiseModelConfig,oracle.pgx.api.internal.mllib.UnsupervisedGraphWiseModelMetadata,UnsupervisedGraphWiseModel>

A variant of Deep Graph Infomax (DGI). See UnsupervisedGraphWiseModelBuilder for documentation of the hyperparameters.

Since:

21.2

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	UnsupervisedGraphWiseModel.UnsupervisedGraphWiseInferenceType

Field Summary

Fields

Modifier and Type	Field and Description
static java.lang.String	ALGORITHM_NAME

Constructor Summary

Constructors

Constructor and Description
UnsupervisedGraphWiseModel(PgxSession session, oracle.pgx.api.internal.Core core, java.util.function.Supplier<java.lang.String> keystorePathSupplier, java.util.function.Supplier<char[]> keystorePasswordSupplier, oracle.pgx.api.internal.mllib.ModelMetadata modelMetadata, java.util.function.BiFunction<PgxSession,oracle.pgx.api.internal.Graph,PgxGraph> graphConstructor)
UnsupervisedGraphWiseModel(PgxSession session, oracle.pgx.api.internal.Core core, java.util.function.Supplier<java.lang.String> keystorePathSupplier, java.util.function.Supplier<char[]> keystorePasswordSupplier, oracle.pgx.api.internal.mllib.UnsupervisedGraphWiseModelMetadata modelMetadata, java.util.function.BiFunction<PgxSession,oracle.pgx.api.internal.Graph,PgxGraph> graphConstructor) This constructor should never be used to get a model.

Method Summary

Modifier and Type	Method and Description
PgxFuture<java.lang.Double>	fitAsync(PgxGraph graph) Trains the GraphWise model on the input graph.
GraphWiseDgiLayerConfig	getDgiLayerConfigs() Gets the configuration object for the dgi layer
UnsupervisedGraphWiseModelConfig.LossFunction	getLossFunction() Gets the loss function
UnsupervisedGnnExplainer	gnnExplainer() Get a GnnExplainer object that can explain this model's predictions.
<ID> PgxFuture<PgxFrame>	inferEmbeddingsAsync(PgxGraph graph, java.lang.Iterable<PgxVertex<ID>> vertices) Infers the embeddings for the specified vertices.
void	store(java.lang.String path, java.lang.String key) Blocking version of storeAsync(String, String).
void	store(java.lang.String path, java.lang.String key, boolean overwrite) Blocking version of storeAsync(String, String).
PgxFuture<java.lang.Void>	storeAsync(java.lang.String path, java.lang.String key) Stores the GraphWise model in the specified path, with encryption.
PgxFuture<java.lang.Void>	storeAsync(java.lang.String path, java.lang.String key, boolean overwrite) Stores the GraphWise model in the specified path, with encryption.

Methods inherited from class oracle.pgx.api.mllib.GraphWiseModel

destroy, destroyAsync, fit, getBatchSize, getConfig, getConvLayerConfigs, getEdgeInputFeatureDim, getEdgeInputPropertyNames, getEmbeddingDim, getInputFeatureDim, getLearningRate, getNumEpochs, getSeed, getTrainingLoss, getVertexInputPropertyNames, inferEmbeddings, isFitted

Methods inherited from class oracle.pgx.api.mllib.Model

close, export

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

ALGORITHM_NAME

public static final java.lang.String ALGORITHM_NAME

See Also:

Constant Field Values

Constructor Detail

UnsupervisedGraphWiseModel

public UnsupervisedGraphWiseModel(PgxSession session,
                                  oracle.pgx.api.internal.Core core,
                                  java.util.function.Supplier<java.lang.String> keystorePathSupplier,
                                  java.util.function.Supplier<char[]> keystorePasswordSupplier,
                                  oracle.pgx.api.internal.mllib.ModelMetadata modelMetadata,
                                  java.util.function.BiFunction<PgxSession,oracle.pgx.api.internal.Graph,PgxGraph> graphConstructor)

UnsupervisedGraphWiseModel

public UnsupervisedGraphWiseModel(PgxSession session,
                                  oracle.pgx.api.internal.Core core,
                                  java.util.function.Supplier<java.lang.String> keystorePathSupplier,
                                  java.util.function.Supplier<char[]> keystorePasswordSupplier,
                                  oracle.pgx.api.internal.mllib.UnsupervisedGraphWiseModelMetadata modelMetadata,
                                  java.util.function.BiFunction<PgxSession,oracle.pgx.api.internal.Graph,PgxGraph> graphConstructor)

This constructor should never be used to get a model. Use UnsupervisedGraphWiseModelBuilder instead.

Parameters:

session - PgxSession to which the model is connected

core - Core to which the model is connected

modelMetadata - Metadata concerning the different hyper-parameters of the GraphWise Model

Since:

21.2

Method Detail

fitAsync

public PgxFuture<java.lang.Double> fitAsync(PgxGraph graph)

Trains the GraphWise model on the input graph.

Specified by:

fitAsync in class GraphWiseModel<UnsupervisedGraphWiseModelConfig,oracle.pgx.api.internal.mllib.UnsupervisedGraphWiseModelMetadata,UnsupervisedGraphWiseModel>

Parameters:

graph - input graph to fit on.

Since:

21.2

getDgiLayerConfigs

public GraphWiseDgiLayerConfig getDgiLayerConfigs()

Gets the configuration object for the dgi layer

Returns:

configuration

Since:

21.2

getLossFunction

public UnsupervisedGraphWiseModelConfig.LossFunction getLossFunction()

Gets the loss function

Returns:

loss function

Since:

21.2

gnnExplainer

public UnsupervisedGnnExplainer gnnExplainer()

Get a GnnExplainer object that can explain this model's predictions.

Returns:

UnsupervisedGnnExplainer object

Since:

22.2

inferEmbeddingsAsync

public <ID> PgxFuture<PgxFrame> inferEmbeddingsAsync(PgxGraph graph,
                                                     java.lang.Iterable<PgxVertex<ID>> vertices)

Infers the embeddings for the specified vertices.

Specified by:

inferEmbeddingsAsync in class GraphWiseModel<UnsupervisedGraphWiseModelConfig,oracle.pgx.api.internal.mllib.UnsupervisedGraphWiseModelMetadata,UnsupervisedGraphWiseModel>

Returns:

PgxFrame containing the embeddings for each vertex.

Since:

21.2

store

public void store(java.lang.String path,
                  java.lang.String key)
           throws java.util.concurrent.ExecutionException,
                  java.lang.InterruptedException

Blocking version of storeAsync(String, String). Calls storeAsync(String, String) and waits for the returned PgxFuture to complete.

Throws:

java.lang.InterruptedException - if the caller thread gets interrupted while waiting for completion.

java.util.concurrent.ExecutionException - if any exception occurred during asynchronous execution. The actual exception will be nested.

Since:

21.2

store

public void store(java.lang.String path,
                  java.lang.String key,
                  boolean overwrite)
           throws java.util.concurrent.ExecutionException,
                  java.lang.InterruptedException

Blocking version of storeAsync(String, String). Calls storeAsync(String, String) and waits for the returned PgxFuture to complete.

Throws:

java.lang.InterruptedException - if the caller thread gets interrupted while waiting for completion.

java.util.concurrent.ExecutionException - if any exception occurred during asynchronous execution. The actual exception will be nested.

Since:

21.2

storeAsync

public PgxFuture<java.lang.Void> storeAsync(java.lang.String path,
                                            java.lang.String key)
                                     throws java.util.concurrent.ExecutionException,
                                            java.lang.InterruptedException

Stores the GraphWise model in the specified path, with encryption.

Parameters:

path - path to store the model

key - the encryption key, or null if no encryption should be used.

Throws:

java.util.concurrent.ExecutionException

java.lang.InterruptedException

Since:

21.2

storeAsync

public PgxFuture<java.lang.Void> storeAsync(java.lang.String path,
                                            java.lang.String key,
                                            boolean overwrite)

Stores the GraphWise model in the specified path, with encryption.

Parameters:

path - path to store the model

key - the encryption key, or null if no encryption should be used.

Since:

21.2