Perform Text classification using DistilBERT
This repository contains a machine learning model for text classification using the DistilBERT model. The model is built using the PyTorch deep learning framework and is trained to classify text data into predefined categories.
Dataset The model is trained on a dataset with 1048575 rows × 4 columns
Requirements The following packages are required to run the model:
torch==1.9.0 transformers==4.10.0 numpy==1.21.2 scikit-learn==1.0.1 The required packages can be installed using the following command: Link to Google Colab Notes
https://colab.research.google.com/drive/1LUTc0sN8QH-UaOdmA9YrHSrazUYvecd9?usp=sharing