Python DataReaderの例

コード例 #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)

コード例 #2

ファイル: test_emf.py プロジェクト: 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()

コード例 #3

ファイル: test_autoencoder.py プロジェクト: 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()

コード例 #4

ファイル: test_mlp.py プロジェクト: 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()

コード例 #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()

コード例 #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)

コード例 #7

ファイル: test_implicit_models.py プロジェクト: 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()

コード例 #8

ファイル: test_emf.py プロジェクト: ludovikcoba/recoxplainer

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

コード例 #9

ファイル: test_autoencoder.py プロジェクト: 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()

コード例 #10

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

コード例 #11

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

コード例 #12

ファイル: test_mlp.py プロジェクト: 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()