CS 294 project using data from Yelp Dataset Challenge