Défi : Implémentation de la Régression Linéaire
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 de régression linéaire simple.
- 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 d'étude. À 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
Merci pour vos commentaires !
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
Can you explain this in simpler terms?
What are some examples related to this topic?
Where can I learn more about this?
Awesome!
Completion rate improved to 5
Défi : Implémentation de la Régression Linéaire
Glissez pour afficher le menu
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 de régression linéaire simple.
- 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 d'étude. À 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
Merci pour vos commentaires !
single