Hi, does this repo also work well with multi-class classification instead of binary classification? Thank you,