Ejemplos de DataReader en Python

Ejemplo n.º 1

    def setUp(self):
        self.als = ALS(**cfg.model.als)

        self.data = DataReader(**cfg.testdata)
        self.data.make_consecutive_ids_in_dataset()
        self.data.binarize()

        self.als.fit(self.data)

Ejemplo n.º 2

Archivo: test_emf.py Proyecto: ludovikcoba/recoxplainer

class EMFTest(unittest.TestCase):
    def setUp(self) -> None:
        self.emf = EMFModel(**cfg.model.emf)
        self.data = DataReader(**cfg.testdata)
        self.data.make_consecutive_ids_in_dataset()

    def test_train_emf(self):
        self.assertTrue(self.emf.fit(self.data))
        recommender = Recommender(self.data, self.emf)
        recommender.recommend_all()

Ejemplo n.º 3

Archivo: test_autoencoder.py Proyecto: ludovikcoba/recoxplainer

class GMFTest(unittest.TestCase):
    def setUp(self) -> None:
        self.autoencoder = ExplAutoencoderTorch(**cfg.model.autoencoder)
        self.data = DataReader(**cfg.testdata)
        self.data.make_consecutive_ids_in_dataset()
        self.data.binarize()

    def test_train_autoencoder(self):
        self.assertTrue(self.autoencoder.fit(self.data))
        recommender = Recommender(self.data, self.autoencoder)
        recommender.recommend_all()

Ejemplo n.º 4

Archivo: test_mlp.py Proyecto: ludovikcoba/recoxplainer

class MLPTest(unittest.TestCase):

    def setUp(self) -> None:
        self.mlp = MLPModel(**cfg.model.mlp)
        self.data = DataReader(**cfg.testdata)
        self.data.make_consecutive_ids_in_dataset()
        self.data.binarize()

    def test_train_mlp(self):
        self.assertTrue(self.mlp.fit(self.data))
        recommender = Recommender(self.data, self.mlp)
        recommender.recommend_all()

Ejemplo n.º 5

class RecommenderImplicitTest(unittest.TestCase):
    def setUp(self):
        self.als = ALS(**cfg.model.als)

        self.data = DataReader(**cfg.testdata)
        self.data.make_consecutive_ids_in_dataset()
        self.data.binarize()

        self.als.fit(self.data)

    def test_train_recommend_als(self):
        recommender = Recommender(self.data, self.als)
        recommender.recommend_all()

Ejemplo n.º 6

class UserItemMatrixTest(unittest.TestCase):

    def setUp(self) -> None:
        self.data = DataReader(**cfg.testdata)
        self.data.make_consecutive_ids_in_dataset()

    def test_user_item_matrix(self):
        user_dict = UserItemDict(self.data.dataset)

        x = self.data.dataset.userId[0]
        y = self.data.dataset.itemId[0]
        v = self.data.dataset.rating[0]
        self.assertEqual(user_dict[x][y], v)

Ejemplo n.º 7

Archivo: test_implicit_models.py Proyecto: ludovikcoba/recoxplainer

class ALSTest(unittest.TestCase):
    def setUp(self):
        self.als = ALS(**cfg.model.als)
        self.bpr = BPR(**cfg.model.bpr)
        self.data = DataReader(**cfg.testdata)
        self.data.make_consecutive_ids_in_dataset()
        self.data.binarize()

    def test_train_als(self):
        self.assertTrue(self.als.fit(self.data))
        recommender = Recommender(self.data, self.als)
        recommender.recommend_all()

    def test_train_bpr(self):
        self.assertTrue(self.bpr.fit(self.data))
        recommender = Recommender(self.data, self.bpr)
        recommender.recommend_all()

Ejemplo n.º 8

Archivo: test_emf.py Proyecto: ludovikcoba/recoxplainer

 def setUp(self) -> None:
     self.emf = EMFModel(**cfg.model.emf)
     self.data = DataReader(**cfg.testdata)
     self.data.make_consecutive_ids_in_dataset()

Ejemplo n.º 9

Archivo: test_autoencoder.py Proyecto: ludovikcoba/recoxplainer

 def setUp(self) -> None:
     self.autoencoder = ExplAutoencoderTorch(**cfg.model.autoencoder)
     self.data = DataReader(**cfg.testdata)
     self.data.make_consecutive_ids_in_dataset()
     self.data.binarize()

Ejemplo n.º 10

 def setUp(self) -> None:
     self.data = DataReader(**cfg.testdata)

Ejemplo n.º 11

 def setUp(self) -> None:
     self.data = DataReader(**cfg.testdata)
     self.data.make_consecutive_ids_in_dataset()

Ejemplo n.º 12

Archivo: test_mlp.py Proyecto: ludovikcoba/recoxplainer

 def setUp(self) -> None:
     self.mlp = MLPModel(**cfg.model.mlp)
     self.data = DataReader(**cfg.testdata)
     self.data.make_consecutive_ids_in_dataset()
     self.data.binarize()