Défi : Implémentation de la Régression Linéaire
Tâche
Swipe to start coding
Vous disposez d'un jeu de données contenant des informations sur le nombre d'heures d'étude des étudiants et leurs scores aux tests correspondants. Votre tâche consiste à entraîner un modèle de régression linéaire sur ces données.
- Convertir ces colonnes en tensors PyTorch, et les redimensionner pour qu'ils soient 2D avec des formes
[N, 1]. - Définir un modèle simple de régression linéaire.
- Utiliser la MSE comme fonction de perte.
- Définir l'
optimizercomme SGD avec un taux d'apprentissage égal à0.01. - Entraîner le modèle de régression linéaire pour prédire les scores aux tests en fonction du nombre d'heures étudiées. À chaque époque :
- Calculer les prédictions sur
X_tensor; - Calculer la perte ;
- Réinitialiser le gradient ;
- Effectuer la rétropropagation ;
- Mettre à jour les paramètres.
- Calculer les prédictions sur
- Accéder aux paramètres du modèle (poids et biais).
Solution
Tout était clair ?
Merci pour vos commentaires !
Section 2. Chapitre 4
single
Demandez à l'IA
Demandez à l'IA
Posez n'importe quelle question ou essayez l'une des questions suggérées pour commencer notre discussion
Awesome!
Completion rate improved to 5
Défi : Implémentation de la Régression Linéaire
Glissez pour afficher le menu
Tâche
Swipe to start coding
Vous disposez d'un jeu de données contenant des informations sur le nombre d'heures d'étude des étudiants et leurs scores aux tests correspondants. Votre tâche consiste à entraîner un modèle de régression linéaire sur ces données.
- Convertir ces colonnes en tensors PyTorch, et les redimensionner pour qu'ils soient 2D avec des formes
[N, 1]. - Définir un modèle simple de régression linéaire.
- Utiliser la MSE comme fonction de perte.
- Définir l'
optimizercomme SGD avec un taux d'apprentissage égal à0.01. - Entraîner le modèle de régression linéaire pour prédire les scores aux tests en fonction du nombre d'heures étudiées. À chaque époque :
- Calculer les prédictions sur
X_tensor; - Calculer la perte ;
- Réinitialiser le gradient ;
- Effectuer la rétropropagation ;
- Mettre à jour les paramètres.
- Calculer les prédictions sur
- Accéder aux paramètres du modèle (poids et biais).
Solution
Tout était clair ?
Merci pour vos commentaires !
Section 2. Chapitre 4
single