add transformData function
Split into train / predict. Support avro