AI Product Classifier – Système intelligent de classification de produits

À propos

AI Product Classifier est un projet personnel de machine learning appliqué, conçu pour répondre à une problématique concrète : structurer automatiquement des données produits non normalisées issues de tickets de caisse ou de bases hétérogènes. Le cœur du projet repose sur le fine-tuning du modèle CamemBERT (HuggingFace Transformers), entraîné de manière supervisée sur un dataset réel de près de 180 000 produits. Le modèle prédit une catégorie parmi plusieurs classes et retourne un score de confiance pour chaque prédiction — y compris sur des libellés bruités comme "Coca cola zero 33cl" ou "Pain complet bio 500g". Sur le plan technique, j'ai géré l'intégralité du pipeline : encodage des labels, entraînement, évaluation par métrique d'accuracy, sauvegarde du modèle pour usage en production et optimisation pour exécution sur CPU et Apple Silicon (MPS). L'architecture est conçue pour être intégrée sous forme de micro-service ou API REST. Ce projet m'a permis d'approfondir concrètement le fine-tuning de modèles de langage et leur mise en production dans un contexte d'analyse de données réelles.