Challenge: Classifying Flowers
Task
Swipe to start coding
Your goal is to train and evaluate a simple neural network using the Iris dataset, which consists of flower measurements and species classification.
- Split the dataset into training and testing sets allocating 20% for the test set and setting random state to
42. - Convert
X_trainandX_testinto PyTorch tensors of typefloat32. - Convert
y_trainandy_testinto PyTorch tensors of typelong. - Define a neural network model by creating the
IrisModelclass. - Implement two fully connected layers and apply the ReLU activation function in the hidden layer.
- Initialize the model with the correct input size, hidden layer size equal to
16, and output size. - Define the loss as cross-entropy loss and the optimizer as Adam with a learning rate of
0.01. - Train the model for 100 epochs by performing forward propagation, computing loss, performing backpropagation, and updating the model's parameters.
- Set the model to evaluation mode after training.
- Disable gradient computation during testing to improve efficiency.
- Compute predictions on the test set using the trained model.
- Determine the predicted class labels based on raw predictions.
Solution
Everything was clear?
Thanks for your feedback!
SectionΒ 1. ChapterΒ 20
single
Ask AI
Ask AI
Ask anything or try one of the suggested questions to begin our chat
Awesome!
Completion rate improved to 5
Challenge: Classifying Flowers
Swipe to show menu
Task
Swipe to start coding
Your goal is to train and evaluate a simple neural network using the Iris dataset, which consists of flower measurements and species classification.
- Split the dataset into training and testing sets allocating 20% for the test set and setting random state to
42. - Convert
X_trainandX_testinto PyTorch tensors of typefloat32. - Convert
y_trainandy_testinto PyTorch tensors of typelong. - Define a neural network model by creating the
IrisModelclass. - Implement two fully connected layers and apply the ReLU activation function in the hidden layer.
- Initialize the model with the correct input size, hidden layer size equal to
16, and output size. - Define the loss as cross-entropy loss and the optimizer as Adam with a learning rate of
0.01. - Train the model for 100 epochs by performing forward propagation, computing loss, performing backpropagation, and updating the model's parameters.
- Set the model to evaluation mode after training.
- Disable gradient computation during testing to improve efficiency.
- Compute predictions on the test set using the trained model.
- Determine the predicted class labels based on raw predictions.
Solution
Everything was clear?
Thanks for your feedback!
SectionΒ 1. ChapterΒ 20
single