private void writeAutoInferredFeature()

in src/main/java/com/amazonaws/services/neptune/profiles/neptune_ml/v2/PropertyGraphTrainingDataConfigWriterV2.java [404:436]

30 lines of code
11 McCabe index (conditional complexity)


    private void writeAutoInferredFeature(PropertySchema propertySchema) throws IOException {

        if (propertySchema.dataType() == DataType.String ||
                propertySchema.dataType() == DataType.Boolean) {
            writeAutoFeature(Collections.singletonList(propertySchema), ImputerTypeV2.none);
        }

        if (propertySchema.dataType() == DataType.Byte ||
                propertySchema.dataType() == DataType.Short ||
                propertySchema.dataType() == DataType.Integer ||
                propertySchema.dataType() == DataType.Long ||
                propertySchema.dataType() == DataType.Float ||
                propertySchema.dataType() == DataType.Double) {
            if (propertySchema.isMultiValue()) {
                writeAutoFeature(Collections.singletonList(propertySchema), ImputerTypeV2.median);
            } else {
                writeNumericalFeature(
                        Collections.singletonList(propertySchema),
                        Norm.min_max,
                        ImputerTypeV2.median);
            }
        }

        if (propertySchema.dataType() == DataType.Date) {
            writeDatetimeFeature(
                    Collections.singletonList(propertySchema),
                    Arrays.asList(
                            DatetimePartV2.year,
                            DatetimePartV2.month,
                            DatetimePartV2.weekday,
                            DatetimePartV2.hour));
        }
    }