public class UnsupervisedEdgeWiseModelBuilder extends EdgeWiseModelBuilder<UnsupervisedEdgeWiseModel,UnsupervisedEdgeWiseModelConfig,UnsupervisedEdgeWiseModelBuilder>
UnsupervisedEdgeWiseModel
The builder can be used to set the configuration of the model and create the model object.Constructor and Description |
---|
UnsupervisedEdgeWiseModelBuilder(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 |
---|---|
UnsupervisedEdgeWiseModel |
build()
Builds the UnsupervisedEdgeWise model with the specified parameters.
|
UnsupervisedEdgeWiseModelBuilder |
setDgiLayerConfig(GraphWiseDgiLayerConfig layerConfig)
Set the dgi layer configurations (See
GraphWiseDgiLayerConfig . |
UnsupervisedEdgeWiseModelBuilder |
setLossFunction(UnsupervisedEdgeWiseModelConfig.LossFunction lossFunction)
Sets the loss function for the algorithm.
|
setEdgeCombinationMethod, setTargetEdgeLabels, setTargetEdgeLabels
setBatchSize, setConvLayerConfigs, setEdgeInputPropertyConfigs, setEdgeInputPropertyNames, setEdgeInputPropertyNames, setEmbeddingDim, setLearningRate, setNormalize, setNumEpochs, setSeed, setShuffle, setStandardize, setVertexInputPropertyConfigs, setVertexInputPropertyNames, setVertexInputPropertyNames, setWeightDecay
public UnsupervisedEdgeWiseModelBuilder(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 UnsupervisedEdgeWiseModel build() throws java.lang.InterruptedException, java.util.concurrent.ExecutionException
build
in class EdgeWiseModelBuilder<UnsupervisedEdgeWiseModel,UnsupervisedEdgeWiseModelConfig,UnsupervisedEdgeWiseModelBuilder>
java.lang.InterruptedException
java.util.concurrent.ExecutionException
public UnsupervisedEdgeWiseModelBuilder 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 UnsupervisedEdgeWiseModelBuilder setLossFunction(UnsupervisedEdgeWiseModelConfig.LossFunction lossFunction)
UnsupervisedEdgeWiseModelConfig.LossFunction
default: UnsupervisedEdgeWiseModelConfig.LossFunction#SIGMOID_CROSS_ENTROPY
lossFunction
- loss function