ML/AI Technical Evaluation

Setup

1. Install dependencies

pip install -r requirements.txt

2. Run the baseline model

The dataset is already included in the repository (compressed). It will automatically decompress on first use.

python train.py

This will generate training plots in the plots/ directory showing loss and accuracy curves over epochs.

Project Structure

• dataset.py - Data loading and preprocessing (auto-decompresses dataset on first use)
• model.py - Neural network model definition
• train.py - Training script