public class UnsupervisedGraphWiseModelBuilder extends GraphWiseModelBuilder<UnsupervisedGraphWiseModel,UnsupervisedGraphWiseModelConfig,UnsupervisedGraphWiseModelBuilder>
UnsupervisedGraphWiseModel
The builder can be used to set the configuration of the model and create the model object.Constructor and Description |
---|
UnsupervisedGraphWiseModelBuilder(PgxSession session, oracle.pgx.api.internal.Core core, java.util.function.Supplier<java.lang.String> keystorePathSupplier, java.util.function.Supplier<char[]> keystorePasswordSupplier, java.util.function.BiFunction<PgxSession,oracle.pgx.api.internal.Graph,PgxGraph> graphConstructor) |
Modifier and Type | Method and Description |
---|---|
UnsupervisedGraphWiseModel |
build()
Builds the UnsupervisedGraphWise model with the specified parameters.
|
UnsupervisedGraphWiseModelBuilder |
setDgiLayerConfig(GraphWiseDgiLayerConfig layerConfig)
Set the dgi layer configurations (See
GraphWiseDgiLayerConfig . |
UnsupervisedGraphWiseModelBuilder |
setLossFunction(UnsupervisedGraphWiseModelConfig.LossFunction lossFunction)
Sets the loss function for the algorithm.
|
setBatchSize, setConvLayerConfigs, setEdgeInputPropertyNames, setEdgeInputPropertyNames, setEmbeddingDim, setLearningRate, setNumEpochs, setSeed, setShuffle, setStandardize, setTargetVertexLabels, setTargetVertexLabels, setVertexInputPropertyNames, setVertexInputPropertyNames, setWeightDecay
public UnsupervisedGraphWiseModelBuilder(PgxSession session, oracle.pgx.api.internal.Core core, java.util.function.Supplier<java.lang.String> keystorePathSupplier, java.util.function.Supplier<char[]> keystorePasswordSupplier, java.util.function.BiFunction<PgxSession,oracle.pgx.api.internal.Graph,PgxGraph> graphConstructor)
public UnsupervisedGraphWiseModel build() throws java.lang.InterruptedException, java.util.concurrent.ExecutionException
build
in class GraphWiseModelBuilder<UnsupervisedGraphWiseModel,UnsupervisedGraphWiseModelConfig,UnsupervisedGraphWiseModelBuilder>
java.lang.InterruptedException
java.util.concurrent.ExecutionException
public UnsupervisedGraphWiseModelBuilder setDgiLayerConfig(GraphWiseDgiLayerConfig layerConfig)
GraphWiseDgiLayerConfig
. You must pass at exactly one prediction layer config (or leave it as default).
default: UnsupervisedGraphWiseModelConfig.DEFAULT_DGI_LAYER_CONFIG
layerConfig
- layer configspublic UnsupervisedGraphWiseModelBuilder setLossFunction(UnsupervisedGraphWiseModelConfig.LossFunction lossFunction)
UnsupervisedGraphWiseModelConfig.LossFunction
default: UnsupervisedGraphWiseModelConfig.LossFunction#SIGMOID_CROSS_ENTROPY
lossFunction
- loss function