Thank you for the suggestion! I am using the method suggested here for training the model: ML-and-Data-Analysis/RoBERTa for text classification.ipynb at master · aramakus/ML-and-Data-Analysis · GitHub 1
It saves the model as two files, a model.pkl and a metrics.pkl file.
Basically, I am trying to predict new labels using this model. Is the dataloader then the right approach?