Mutli class classification using ERNIE

Question

I\'m following this tutorial here to implement ERNIE 1.0, which runs perfectly fine for a 2 label classification task such as the IMDB movie review dataset.

However,