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, setWeightDecaypublic abstract Model build() throws java.lang.InterruptedException, java.util.concurrent.ExecutionException
java.lang.InterruptedExceptionjava.util.concurrent.ExecutionExceptionpublic Self setTargetVertexLabels(java.util.List<java.lang.String> targetVertexLabels)
targetVertexLabels - list of label namespublic Self setTargetVertexLabels(java.lang.String... targetVertexLabels)
targetVertexLabels - label names