classes.py

#!/usr/bin/env python
# coding: utf-8

# In[ ]:


from util import random_vector
class user:
    def __init__(self, userid):
        self.userid = userid
        self.movies_train = dict()
        self.movies_test = dict()
        self.movies_all = dict()
        self.factor = random_vector()
    
class movie:
    def __init__(self, movieid, rating=0, title=None, genres=None):
        self.movieid = movieid
        self.rating = rating
        self.title = title
        self.genres = genres
        self.factor = random_vector()

class ret:
    def __init__(self):
        self.userid = None
        self.movieid = None
        self.isuser = True
        self.retvalue = []
    
class usermovie:
    def __init__(self):
        self.userid = None
        self.movieid = None
        self.rating = 0