public abstract class GraphWiseModelBuilder<Model extends GraphWiseModel,Config extends GraphWiseModelConfig,Self extends GraphWiseModelBuilder> extends WiseModelBuilder<Config,Self>
SupervisedGraphWiseModelBuilder
instead of this.Constructor and Description |
---|
GraphWiseModelBuilder() |
Modifier and Type | Method and Description |
---|---|
abstract Model |
build()
Builds the GraphWise model with the specified parameters.
|
Self |
setTargetVertexLabels(java.util.List<java.lang.String> targetVertexLabels)
Set the target vertex labels for the algorithm.
|
Self |
setTargetVertexLabels(java.lang.String... targetVertexLabels)
Set the target vertex labels for the algorithm.
|
setBatchSize, setConvLayerConfigs, setEdgeInputPropertyConfigs, setEdgeInputPropertyNames, setEdgeInputPropertyNames, setEmbeddingDim, setLearningRate, setNormalize, setNumEpochs, setSeed, setShuffle, setStandardize, setVertexInputPropertyConfigs, setVertexInputPropertyNames, setVertexInputPropertyNames, setWeightDecay
public abstract Model build() throws java.lang.InterruptedException, java.util.concurrent.ExecutionException
java.lang.InterruptedException
java.util.concurrent.ExecutionException
public Self setTargetVertexLabels(java.util.List<java.lang.String> targetVertexLabels)
targetVertexLabels
- list of label namespublic Self setTargetVertexLabels(java.lang.String... targetVertexLabels)
targetVertexLabels
- label names