Skip to content

Bogazici-SWE546-Spring2016/SWE546_Assignment4

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 

Repository files navigation

SWE546_Assignment4

MovieLens Recommendation System

http://grouplens.org/datasets/movielens/

Using the MovieLens 1M dataset, and using only the rankings, implement a recommendation sytem using SGD.

import pandas as pd

rnames = ['user_id', 'movie_id', 'rating', 'timestamp'] ratings = pd.read_table('data/ml-1m/ratings.dat', sep='::', header=None, names=rnames)

The output of your program must be the highest estimated rankings for the first 10 movies of the first 20 users

Modify the program for mini batch and classical gradient descent.

Compare the running times and solution quality as a function of the batch size. SGD is when batchsize = 1, Gradient descent is batchsize = Number of observed entries in the matrix.

Submit a short report with the plots of your results.

About

MovieLens with Stochastic Gradient Descent

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published