Retour aux cas pratiques
PyTorchAvancé

Mise en place d’un entraînement avec gestion du déséquilibre des classes

Scénario

Contexte : Un modèle de classification ignore la classe minoritaire. Le Problème : Les données sont fortement déséquilibrées. Contraintes : Utiliser pondération des classes dans CrossEntropyLoss ou sur-échantillonnage. Évaluer via F1-score macro. Livrable attendu : Implémenter un entraînement tenant compte du déséquilibre et comparer les performances avant/après ajustement.

Compétences ciblées

imbalanced-dataclassificationloss-functionevaluationmodel-training
    Mise en place d’un entraînement avec gestion du déséquilibre des classes — mise en situation PyTorch | CodeDuel