Package oracle.pgx.api.mllib

Class GraphWiseModelBuilder<Model extends GraphWiseModel,Config extends GraphWiseModelConfig,Self extends GraphWiseModelBuilder>

java.lang.Object

oracle.pgx.api.mllib.WiseModelBuilder<Config,Self>

oracle.pgx.api.mllib.GraphWiseModelBuilder<Model,Config,Self>

Direct Known Subclasses:

SupervisedGraphWiseModelBuilder, UnsupervisedAnomalyDetectionGraphWiseModelBuilder, UnsupervisedGraphWiseModelBuilder

public abstract class GraphWiseModelBuilder<Model extends GraphWiseModel,Config extends GraphWiseModelConfig,Self extends GraphWiseModelBuilder>
extends WiseModelBuilder<Config,Self>

Abstract builder class for GraphWise models. Use SupervisedGraphWiseModelBuilder instead of this.

Since:

19.4

Constructor Summary

Constructors

Constructor	Description
GraphWiseModelBuilder()

Method Summary

Modifier and Type	Method	Description
abstract Model	build()	Builds the GraphWise model with the specified parameters.
Self	setTargetVertexLabels(String... targetVertexLabels)	Set the target vertex labels for the algorithm.
Self	setTargetVertexLabels(List<String> targetVertexLabels)	Set the target vertex labels for the algorithm.

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

setBatchSize, setConvLayerConfigs, setEdgeInputPropertyConfigs, setEdgeInputPropertyNames, setEdgeInputPropertyNames, setEmbeddingDim, setEnableAccelerator, setLearningRate, setNormalize, setNumEpochs, setSeed, setShuffle, setStandardize, setValidationConfig, setVertexInputPropertyConfigs, setVertexInputPropertyNames, setVertexInputPropertyNames, setWeightDecay

Methods inherited from class java.lang.Object

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

Constructor Details

GraphWiseModelBuilder

public GraphWiseModelBuilder()

Method Details

setTargetVertexLabels

public Self setTargetVertexLabels(List<String> targetVertexLabels)

Set the target vertex labels for the algorithm. Only the related vertices need to have the target property. Training and inference will be done on the vertices with those labels

Parameters:

targetVertexLabels - list of label names

Returns:

this

Since:

22.2

setTargetVertexLabels

public Self setTargetVertexLabels(String... targetVertexLabels)

Set the target vertex labels for the algorithm. Only the related vertices need to have the target property.

Parameters:

targetVertexLabels - label names

Returns:

this

Since:

22.2

build

public abstract Model build()
                     throws InterruptedException,
ExecutionException

Builds the GraphWise model with the specified parameters.

Returns:

GraphWise model

Throws:

InterruptedException

ExecutionException

Since:

19.4