Thank you so much for your reply! I looked into the update of your github repo, yet found the update was about multi-class classification, instead of multi-label classification. I would like to use your project to classify the multi-label problems like toxic comment (https://www.kaggle.com/c/jigsaw-toxic-comment-classification-challenge), in which a certain text can be assigned to more than one labels. Would you please also implement this function? Thanks a lot!