Hi Ratikant,

Yes, feel free to use it. For multi-classification tasks, you need to modify the return value of get_labels() function (depends on the DataProcessor class that you used). Change it from

def get_labels(self):
    return ["0", "1"]

to the following (example using 5 classes):

def get_labels(self):
    return ["0", "1", "2", "3", "4"]

Best regards,

Wai Foong