CnD_Classification/README.md

975 B

Cats and Dogs Classification

This project is meant as a way to gradually bring improvements on the bayesian network until it is as good as I can make it. All papers that I use will be featured in the README. There will also be a report on how the project went and what I learned and improved on.

Table of Contents

Objectives 🎯

  • Refactor the code
  • Generate some graphs to visualize the data
  • [ X ] Make a CLI
  • [ X ] Pretrain some models

Requirements 📋

To run the projet you need the following requirements:

  • Python 3.12
  • venv

Running the project 🚀

python -m venv .venv
source .venv/bin/activate
pip install requirements.txt

python train.py # If you want to train your own model
python main.py

Development 🔨

Citations 📝