How to apply the Tensorflow tutorial for BERT Sentiment Analysis for multi class problem

Question

real beginner here.

I found this Tutorial (https://www.tensorflow.org/tutorials/text/classify_text_with_bert) and tried to change the data input to another dataset with 3