--- library_name: transformers license: mit datasets: - heegyu/news-category-dataset language: - en metrics: - accuracy pipeline_tag: text-classification --- # Model Card for Model ID This model is a finetuning of the distilbert/distilbert-base-uncased llm for the news-category-dataset. ## Model Details Training took ~1 hour on google colab using an L4 GPU. For the dataset presenting 42 labels, it provides the following accuracies: {'eval_loss': 1.003881573677063, 'eval_accuracy': 0.7176901049952275, } ### Model Description This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated. - **Developed by:** [Kirkos27] - **Model type:** Text Classification - **Language(s) (NLP):** English - **Finetuned from model :** distilbert/distilbert-base-uncased