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Β 3. ChapterΒ 4
single
Ask AI
Ask AI
Ask anything or try one of the suggested questions to begin our chat
Suggested prompts:
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
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Β 3. ChapterΒ 4
single