Udfordring: Implementering af Lineær Regression
Opgave
Swipe to start coding
Du får et datasæt, der indeholder information om antallet af timer, som studerende har læst, samt deres tilsvarende testresultater. Din opgave er at træne en lineær regressionsmodel på disse data.
- Konverter disse kolonner til PyTorch-tensorer, og omform dem, så de er 2D med form
[N, 1]
. - Definér en simpel lineær regressionsmodel.
- Brug MSE som tab-funktion.
- Definér
optimizer
som SGD med en læringsrate på0.01
. - Træn den lineære regressionsmodel til at forudsige testresultater baseret på antallet af læste timer. Ved hver epoke:
- Beregn forudsigelser på
X_tensor
; - Beregn tabet;
- Nulstil gradienten;
- Udfør bagudpassage;
- Opdater parametrene.
- Beregn forudsigelser på
- Tilgå modellens parametre (vægte og bias).
Løsning
Var alt klart?
Tak for dine kommentarer!
Sektion 2. Kapitel 4
single
Spørg AI
Spørg AI
Spørg om hvad som helst eller prøv et af de foreslåede spørgsmål for at starte vores chat
Awesome!
Completion rate improved to 5
Udfordring: Implementering af Lineær Regression
Stryg for at vise menuen
Opgave
Swipe to start coding
Du får et datasæt, der indeholder information om antallet af timer, som studerende har læst, samt deres tilsvarende testresultater. Din opgave er at træne en lineær regressionsmodel på disse data.
- Konverter disse kolonner til PyTorch-tensorer, og omform dem, så de er 2D med form
[N, 1]
. - Definér en simpel lineær regressionsmodel.
- Brug MSE som tab-funktion.
- Definér
optimizer
som SGD med en læringsrate på0.01
. - Træn den lineære regressionsmodel til at forudsige testresultater baseret på antallet af læste timer. Ved hver epoke:
- Beregn forudsigelser på
X_tensor
; - Beregn tabet;
- Nulstil gradienten;
- Udfør bagudpassage;
- Opdater parametrene.
- Beregn forudsigelser på
- Tilgå modellens parametre (vægte og bias).
Løsning
Var alt klart?
Tak for dine kommentarer!
Awesome!
Completion rate improved to 5Sektion 2. Kapitel 4
single