/Machine-Learning

Work from CS 4990 special topics course on machine learning

Primary LanguageJupyter Notebook

1. Linear regression

Just some basic linear regression implemented using scikit-learn as well as on our own using gradient descent

Data set

https://www.kaggle.com/gregorut/videogamesales

2. Emotion classification

Emotion recognition using CNNs. Attempt 1 scored a 83.5%, Attempt 2 scored a 84.0% in kaggle competition.

Data set

https://www.kaggle.com/c/cs4990-assignment-2-option-3-facial-expression/data

Model

VGG19 inspired model

  • Layers: 64 - 128 - 256 - 512 - 1024, 4096 full connected

3. Hair segmentation

Hair segmentation using Unets. Best result was 78.9%.

Data set

https://www.kaggle.com/c/cs4990-assignment-3-option-1-hair-segmentation/data

Model

Medical processing inspired Unet

  • Layers: 64 - 128 - 256 - 512 - 1024 - 512 - 256 - 128 - 64

4. Spam detection

A practice project. Recognizing spam text messages using a recurrent neural net.

Data set

https://www.kaggle.com/uciml/sms-spam-collection-dataset

Tutorial

https://www.kaggle.com/kentata/rnn-for-spam-detection/notebook