Supervised Learning Projects




This project aims to develop an AI engine that teaches people how to sort trash with computer vision techniques.

I used a pre-trained residual convolutional neural network (resnet50) and, through transfer learning, trained the last layer with a dataset of 2,500 images of trash, obtaining an accuracy of 91% in the test set. Then, I implemented a hardware prototype to turn on LED lights according to the predicted category of trash.

Created at: May 14, 2020              Last modified: May 14, 2020




In this project, I tried to predict rental prices of Airbnb listings from features related to the properties, such as amenities, rooms and location.

I trained different regression models (Linear Regression, Decision Tree and Random Forest) using the python scikit-learn library, and then I comapred the performance of them through cross-validation.

Created at: May 19, 2020              Last modified: June 4, 2020