Beginner-friendly project demonstrating how to build, train, and evaluate a multi-class classification model using PyTorch.
pip install torch torchvision matplotlib scikit-learnFor feedback or questions, contact: [email protected]
-
Notifications
You must be signed in to change notification settings - Fork 0
Multi-class Classification with PyTorch This project trains a simple neural network on a synthetic 4-class dataset created with make_blobs. The model uses two hidden layers and is trained with CrossEntropyLoss and SGD. Decision boundaries are visualized, and test accuracy is reported.
License
barkinadiguzel/Multiclass-Classification-model
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Multi-class Classification with PyTorch This project trains a simple neural network on a synthetic 4-class dataset created with make_blobs. The model uses two hidden layers and is trained with CrossEntropyLoss and SGD. Decision boundaries are visualized, and test accuracy is reported.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published