Class GraphWiseModelConfig

    • Constructor Detail

      • GraphWiseModelConfig

        public GraphWiseModelConfig​(int batchSize,
                                    int numEpochs,
                                    double learningRate,
                                    double weightDecay,
                                    int embeddingDim,
                                    java.lang.Integer seed,
                                    GraphWiseBaseConvLayerConfig[] convLayerConfigs,
                                    boolean standardize,
                                    boolean normalize,
                                    boolean shuffle,
                                    java.util.List<java.lang.String> vertexInputPropertyNames,
                                    java.util.List<java.lang.String> edgeInputPropertyNames,
                                    java.util.Map<java.lang.String,​InputPropertyConfig> vertexInputPropertyConfigs,
                                    java.util.Map<java.lang.String,​InputPropertyConfig> edgeInputPropertyConfigs,
                                    oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig,
                                    boolean fitted,
                                    double trainingLoss,
                                    int vertexInputFeatureDim,
                                    int edgeInputFeatureDim,
                                    java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets,
                                    GraphWiseBaseModelConfig.Backend backend,
                                    GraphWiseModelConfig.GraphConvModelVariant variant,
                                    boolean enableAccelerator,
                                    GraphWiseValidationConfig validationConfig)
      • GraphWiseModelConfig

        public GraphWiseModelConfig​(GraphWiseModelConfig source,
                                    oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
    • Method Detail

      • getTargetVertexLabelSets

        public java.util.List<java.util.Set<java.lang.String>> getTargetVertexLabelSets()
      • setTargetVertexLabelSets

        public final void setTargetVertexLabelSets​(java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets)
      • setTargetVertexLabels

        public void setTargetVertexLabels​(java.util.List<java.lang.String> targetVertexLabels)