Class SupervisedGraphWiseModelConfig

    • Constructor Detail

      • SupervisedGraphWiseModelConfig

        public SupervisedGraphWiseModelConfig()
      • SupervisedGraphWiseModelConfig

        public SupervisedGraphWiseModelConfig​(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,
                                              java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets,
                                              boolean fitted,
                                              double trainingLoss,
                                              int inputFeatureDim,
                                              int edgeInputFeatureDim,
                                              oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction lossFunction,
                                              LossFunction lossFunctionClass,
                                              BatchGenerator batchGenerator,
                                              GraphWisePredictionLayerConfig[] predictionLayerConfigs,
                                              java.lang.String vertexTargetPropertyName,
                                              LabelMaps labelMaps,
                                              GraphWiseBaseModelConfig.Backend backend,
                                              GraphWiseModelConfig.GraphConvModelVariant variant,
                                              boolean enableAccelerator,
                                              GraphWiseValidationConfig validationConfig)
      • SupervisedGraphWiseModelConfig

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

      • getNumClasses

        public int getNumClasses()
      • getVertexTargetPropertyName

        public java.lang.String getVertexTargetPropertyName()
      • setVertexTargetPropertyName

        public final void setVertexTargetPropertyName​(java.lang.String vertexTargetPropertyName)
      • getLossFunctionClass

        public LossFunction getLossFunctionClass()
      • setLossFunctionClass

        public final void setLossFunctionClass​(LossFunction lossFunction)
      • setBatchGenerator

        public final void setBatchGenerator​(BatchGenerator batchGenerator)
      • getClassMap

        public java.util.Map<?,​java.lang.Integer> getClassMap()
      • setClassMap

        public final void setClassMap​(java.util.Map<?,​java.lang.Integer> classMap)
      • setClassWeights

        public final void setClassWeights​(java.util.Map<?,​java.lang.Float> classWeights)
      • getClassWeights

        public java.util.Map<?,​java.lang.Float> getClassWeights()
      • setLabelType

        public final void setLabelType​(PropertyType labelType)
      • getLabelMaps

        public LabelMaps getLabelMaps()
      • setLabelMaps

        public final void setLabelMaps​(LabelMaps labelMaps)