E-commerce item category prediction using item description

Hi all,

Is there a way I can use fine tune a text classification model to give me categories associated with a product. The data I have consists of “product description” and “product category”.

How can I modify the existing sentiment classification models to output probabilities into a set of categories instead of “positive” and “negative” sentiment?

Any help would be much appreciated. Thanks!