I don't think you need to define format_func here, that's just for converting to one-hot vectors.