Can this repo be used in multi-classification task?

[youngboy52]

Hi, I see that the prepare_data.py loads data from directory, so is it limited to single classification task? if not, how to modify it in order to use it in multi-classification task? Look forward to your reply!