Example #1
0
 def __init__(self, reader=None, data=None):
     Dataset.__init__(self, reader)
     self.ratings_file = None
     self.n_folds = 5
     self.shuffle = True
     if data.shape[-1] == 3:
         data = self.change_format(data)
     self.raw_ratings = data
Example #2
0
    def __init__(self, host, database, charset, reader=None):

        Dataset.__init__(self, reader)
        self.host = host
        self.database = database
        self.charset = charset
        self.n_folds = 5
        self.shuffle = True
        self.result = None
        self.raw_ratings = None
        self.my_connect = None