Cant you use the new  nlp library to setup the dataset? I haven’t tried creating one on a custom data as yet and to use it directly with trainer but I think they provider wrappers for CSV which should help.