Construction d’un premier modèle de classification avec PyTorch

Scénario

Contexte : Une startup souhaite prédire si un client va se désabonner à partir de données numériques simples. Le Problème : Aucun modèle n’a encore été développé avec PyTorch et l’équipe débute avec cette bibliothèque. Contraintes : Définir un réseau simple avec torch.nn.Module. Utiliser une couche linéaire ou deux couches Fully Connected. Séparer train et test. Utiliser une fonction de perte adaptée à la classification binaire. Livrable attendu : Implémenter un pipeline complet incluant définition du modèle, boucle d’entraînement, calcul de la perte et évaluation sur un jeu de test.

Compétences ciblées