NLP, Python, Transformers, Data Science, Linear Regression.
Load the model Here is the most important step. According to Hugging Face’s library, when we load the pre-trained models from the Hugging Face API, setting the num_labels to 1 for the AutoModelForSequenceClassification will trigger the linear regression and use MSELoss() as the loss function