NLP, Python, Transformers, Data Science, Linear Regression.
R2 (0.6422) and RMSE (0.6589) were attained at epoch 2. And here we go, the last step: let’s test the fine-tuned model on the test dataset. The output is shown below. I would say the results are decent compared to the model’s performance on the validation dataset. For the full code script: