package com.javadeeplearningcookbook.app; import org.datavec.api.transform.TransformProcess; import org.datavec.api.transform.schema.Schema; import org.datavec.api.transform.transform.doubletransform.ConvertToDouble; import java.util.Arrays; public class SerializationExample { public static void main(String[] args) { Schema schema = new Schema.Builder() .addColumnsString("Name", "Subject") .addColumnInteger("Score") .addColumnCategorical("Grade", Arrays.asList("A","B","C","D")) .addColumnInteger("Passed").build(); TransformProcess transformProcess = new TransformProcess.Builder(schema) .removeColumns("Name") .transform(new ConvertToDouble("Score")) .categoricalToInteger("Grade").build(); String json = transformProcess.toJson(); System.out.println(json); String yaml = transformProcess.toYaml(); System.out.println(yaml); } }