Add uniform max len for padding for predictions

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