Model Card for Multi-label Text Classification Model for Toxic Comment Dataset
Model Description
Please refer to the training method, details and training code below:
- Repository: https://github.com/Maker6666/Uncertainty-Aware_Text_Classification/
- Paper: Uncertainty-Aware Sentence Embedding in Text Classification
Uses
- Download the model files in this repository to your local machine.
- Prepare the Toxic Comment test dataset.
- Load the local model and perform testing.