SDSC4016 Fundamentals of Machine Learning II

SDSC4016 Homework 2:

• On Kaggle

Description

Solve an image classification problem with CNN.

Getting Started

Dependencies

• Python
  • Python 3.10+
  • Jupyter
  • pytorch
  • torchvision

Install mini-conda and mamba

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash ./Miniconda3-latest-Linux-x86_64.sh
conda install mamba -n base -c conda-forge

Set up conda environment

mamba create -n 4016hw2
mamba activate 4016hw2

Installing dependencies

# conda or mamba
mamba install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
mamba install -c conda-forge Jupyter ipykernel
mamba install -c conda-forge pandas numpy matplotlib scikit-learn tqdm

Code

Baseline

Modified

Tested Result on Kaggle

Results on Kaggle

Final Score

• Public: 0.70282
• Private: 0.71159