Update gene classification example to create directory after training arguments are defined

Subclass collator for cell classification

Add data collator for gene classification and example for gene classification