Text Classification
Transformers
Safetensors
English
bert
Inference Endpoints

classification or regression? how to choose.

#7
by DaleMeng - opened

Hello,
According to the code in https://github.com/huggingface/cosmopedia/blob/main/classification/train_edu_bert.py, I find that the model is trained based on a regression model instead of a 6-labels classification model. I wonder if there is any deep insight to do it in a regression way? or have you compared it to the classification way?

@DaleMeng I wonder the same, have you tried to use classification loss on your task?

Sign up or log in to comment