These examples are designed to help you get over the hurdle of learning scikit-learn
and pytorch
modules to implement machine learning models for COMP4702.
This repository covers examples like:
- Decision Trees
- kNN Models
- Linear Regression
- Cross-Validation
This package assumes you have a miniconda
environment installed. If you do not, please see minconda.
conda create -n comp4702_examples python=3.10
conda activate comp4702_examples
conda install numpy matplotlib pandas
conda install scikit-learn
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia