How to apply for multi-class classification

[kebitmatf]

Hi,
I'm using built-in models (e.g., DLRM) with binary classification in the examples with movielens dataset. However, the recommender system should be multi-class classification. I tried but cannot find a way to run the models with the multi-class classification. Could anyone give me a guideline or an example?

This is a tutorial that I tried with binary classification. I cannot change it to mutl-class classification:
https://nvidia-merlin.github.io/models/v0.5.0/README.html#notebook-examples-and-tutorials