/CuML

CS-221 Course Project

Primary LanguageCudaMIT LicenseMIT

CuML

License: MIT

This repo contains code for various ML/DL algorithms/models implemented using CUDA as a semester project for CSN-221: Computer Architecture

Repository Structure

  • CuCNN: Contains a simple CNN implemented using CUDA on the MNIST dataset.

  • CuDecisionTree: Contains the Decision Tree algorithm implemented using CUDA.

  • CuKNN: Contains the k-Nearest Neighbour algorithm implemented using CUDA.

  • CuSVD: Contains the SVD algorithm implemented using CUDA.

  • CuSVM: Contains a simple SVM implemented using CUDA.

Setup

  • Clone the repo using:
! git clone https://github.com/praeclarumjj3/CuML.git

Check README.md of each algorithm to see the setup instructions.

Contributors

  • CuCNN: Jitesh Jain

  • CuSVD: Anurag Bansal

  • CuKNN: Anushka Singh

  • CuSVM: Hardik Thami

  • CuDecisionTree: Anshika Mittal