Målet är att träna och utvärdera ett enkelt neuralt nätverk med hjälp av Iris-datasetet, som innehåller blomsters mätvärden och artklassificering.

1. Dela upp datasetet i tränings- och testuppsättningar där 20% används för test och slumpmässigt tillstånd sätts till 42.
2. Konvertera X_train och X_test till PyTorch-tensorer av typen float32.
3. Konvertera y_train och y_test till PyTorch-tensorer av typen long.
4. Definiera en modell för neuralt nätverk genom att skapa klassen IrisModel.
5. Implementera två fullt anslutna lager och använd ReLU-aktiveringsfunktionen i det dolda lagret.
6. Initiera modellen med korrekt indata-storlek, dold lagerstorlek lika med 16 och utdata-storlek.
7. Definiera förlusten som korsentropiförlust och optimizern som Adam med inlärningshastighet 0.01.
8. Träna modellen i 100 epoker genom att utföra framåtriktad spridning, beräkna förlust, utföra bakåtriktad spridning och uppdatera modellens parametrar.
9. Sätt modellen i utvärderingsläge efter träning.
10. Inaktivera gradientberäkning under testning för att förbättra effektiviteten.
11. Beräkna prediktioner på testuppsättningen med den tränade modellen.
12. Bestäm de förutsagda klassetiketterna baserat på råa prediktioner.