CoolFace
Modelpublic

miguelonana/camembert-bank-post-classifier

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes24downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

camembert-bank-post-classifier

It classifies posts on an internal social media in 7 categories :

  • —Annonce RH & Vie de l'Entreprise
  • —Communication de la Direction & Stratégie
  • —Formation & Développement
  • —Information & Alerte Pratique
  • —Projet, Produit & Innovation
  • —RSE & Engagement Citoyen
  • —Événement Externe & Partenariat

Model description

This model is a fine-tuned version of camembert-base on an unknown dataset.

Intended uses & limitations

Classifying posts under 7 categories.

Training and evaluation data

It achieves the following results on the evaluation set:

  • —Loss: 0.5783
  • —Accuracy: 0.8682
  • —F1: 0.8663

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 2e-05
  • —trainbatchsize: 16
  • —evalbatchsize: 16
  • —seed: 42
  • —optimizer: Use OptimizerNames.ADAMWTORCHFUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • —lrschedulertype: linear
  • —num_epochs: 8

Training results

Training LossEpochStepValidation LossAccuracyF1
1.5571.01241.23810.71810.6649
1.04152.02480.88300.83160.8254
0.62583.03720.67250.85600.8523
0.47234.04960.61030.85400.8516
0.30925.06200.59210.86000.8583
0.22446.07440.57220.86410.8624
0.14827.08680.57190.86410.8626
0.14938.09920.57830.86820.8663

Framework versions

  • —Transformers 4.56.1
  • —Pytorch 2.8.0+cu126
  • —Datasets 4.0.0
  • —Tokenizers 0.22.0