in src/main/java/com/amazonaws/services/neptune/profiles/neptune_ml/v1/PropertyGraphTrainingDataConfigWriterV1.java [169:192]
private void writeNodeFeatures(Label label, Collection<PropertySchema> propertySchemas, LabelSchema labelSchema) throws IOException {
boolean arrayStartHasBeenWritten = false;
for (PropertySchema propertySchema : propertySchemas) {
String column = propertySchema.nameWithoutDataType();
if (!config.isNodeClassificationPropertyForNode(label, column)) {
if (!arrayStartHasBeenWritten) {
generator.writeArrayFieldStart("features");
arrayStartHasBeenWritten = true;
}
if (!config.hasNodeFeatureOverrideForNodeProperty(label, column)) {
writeNodeFeature(label, propertySchema, labelSchema);
}
}
}
for (FeatureOverrideConfigV1 featureOverride : config.getNodeFeatureOverrides(label)) {
writeNodeFeatureOverride(label, featureOverride, propertySchemas, labelSchema);
}
if (arrayStartHasBeenWritten) {
generator.writeEndArray();
}
}