https://github.com/memudualimatou/hyperparameter-tuning-with-pytorch
https://github.com/memudualimatou/hyperparameter-tuning-with-pytorch
Last synced: 7 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/memudualimatou/hyperparameter-tuning-with-pytorch
- Owner: memudualimatou
- Created: 2024-03-04T23:18:26.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2024-04-03T15:33:58.000Z (over 1 year ago)
- Last Synced: 2025-01-21T22:43:15.679Z (9 months ago)
- Language: Jupyter Notebook
- Size: 43 KB
- Stars: 1
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
This repository contains a notebook showcasing the process of tuning hyperparameters for a Convolutional Neural Network (CNN) image classification model created using PyTorch. The central component of this implementation is a library known as **skorch**.
The dataset is available through this [link](https://zindi.africa/competitions/sbtic-animal-classification/data), from a Zindi Africa hackathon. This data is composed of images of two different animal categories, elephant and zebra. The goal is to build an image classifaction model.
The content of this repository is to demonstrate the effect of hyperparameter tuning on neural network models.
Wishing you fruitful learning!