https://github.com/televisionninja/basic-multiclass-classification-convolutional-neural-network

A basic multiclass classification convolutional neural network using PyTorch
https://github.com/televisionninja/basic-multiclass-classification-convolutional-neural-network

ai convolutional-neural-networks machine-learning multiclass-classification pytorch

Last synced: 6 months ago
JSON representation

A basic multiclass classification convolutional neural network using PyTorch

• Host: GitHub
• URL: https://github.com/televisionninja/basic-multiclass-classification-convolutional-neural-network
• Owner: TelevisionNinja
• License: mit
• Created: 2023-08-12T17:06:51.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-08-12T17:29:24.000Z (over 2 years ago)
• Last Synced: 2025-01-21T03:45:00.557Z (12 months ago)
• Topics: ai, convolutional-neural-networks, machine-learning, multiclass-classification, pytorch
• Language: Python
• Homepage:
• Size: 154 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# Basic-Multiclass-Classification-Convolutional-Neural-Network

A basic multiclass classification convolutional neural network using PyTorch

# Dependencies

- pytorch

- matplotlib

- mlxtend

- torchmetrics

- tqdm

# Command to run the model

```bash

python main.py

```

# Results

![results](./images/results.png)

![matrix](./images/confusion%20matrix.png)

![loss](./images/loss.png) ![accuracy](./images/accuracy.png)