maker666/Toxic_Comment

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.