private void writeNodeFeature()

in src/main/java/com/amazonaws/services/neptune/profiles/neptune_ml/v1/PropertyGraphTrainingDataConfigWriterV1.java [194:212]

16 lines of code
9 McCabe index (conditional complexity)


    private void writeNodeFeature(Label label, PropertySchema propertySchema, LabelSchema labelSchema) throws IOException {

        if (propertySchema.dataType() == DataType.Float ||
                propertySchema.dataType() == DataType.Double) {
            writeNumericalNodeFeature(label, Collections.singletonList(propertySchema), Norm.min_max, labelSchema);
        }

        if (propertySchema.dataType() == DataType.Byte ||
                propertySchema.dataType() == DataType.Short ||
                propertySchema.dataType() == DataType.Integer ||
                propertySchema.dataType() == DataType.Long) {
            writeNumericalNodeFeature(label, Collections.singletonList(propertySchema), Norm.min_max, labelSchema);
        }

        if (propertySchema.dataType() == DataType.String ||
                propertySchema.dataType() == DataType.Boolean) {
            writeCategoricalNodeFeature(label, Collections.singletonList(propertySchema));
        }
    }