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 étudiées par les é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.
- Convertissez ces colonnes en tensors PyTorch et redimensionnez-les pour qu'ils soient 2D avec des formes
[N, 1]. - Définissez un modèle simple de régression linéaire.
- Utilisez la MSE comme fonction de perte.
- Définissez l'
optimizercomme SGD avec un taux d'apprentissage égal à0.01. - Entraînez 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 :
- Calculez les prédictions sur
X_tensor; - Calculez la perte ;
- Réinitialisez le gradient ;
- Effectuez la rétropropagation ;
- Mettez à jour les paramètres.
- Calculez les prédictions sur
- Accédez aux paramètres du modèle (poids et biais).
Solution
Tout était clair ?
Merci pour vos commentaires !
Section 1. Chapitre 14
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
Génial!
Completion taux amélioré à 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 étudiées par les é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.
- Convertissez ces colonnes en tensors PyTorch et redimensionnez-les pour qu'ils soient 2D avec des formes
[N, 1]. - Définissez un modèle simple de régression linéaire.
- Utilisez la MSE comme fonction de perte.
- Définissez l'
optimizercomme SGD avec un taux d'apprentissage égal à0.01. - Entraînez 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 :
- Calculez les prédictions sur
X_tensor; - Calculez la perte ;
- Réinitialisez le gradient ;
- Effectuez la rétropropagation ;
- Mettez à jour les paramètres.
- Calculez les prédictions sur
- Accédez aux paramètres du modèle (poids et biais).
Solution
Tout était clair ?
Merci pour vos commentaires !
Section 1. Chapitre 14
single