├── .github └── workflows │ ├── python-publish.yml │ └── python-test.yml ├── .gitignore ├── LICENSE ├── PyTorchCML ├── __init__.py ├── adaptors │ ├── BaseAdaptor.py │ ├── MLPAdaptor.py │ └── __init__.py ├── evaluators │ ├── BaseEvaluator.py │ ├── UserwiseEvaluator.py │ └── __init__.py ├── losses │ ├── BaseLoss.py │ ├── LogitPairwiseLoss.py │ ├── MSEPairwiseLoss.py │ ├── MinTripletLoss.py │ ├── RelevancePairwiseLoss.py │ ├── SumTripletLoss.py │ └── __init__.py ├── models │ ├── BaseEmbeddingModel.py │ ├── CollaborativeMetricLearning.py │ ├── MatrixFactorization.py │ └── __init__.py ├── regularizers │ ├── BaseRegularizer.py │ ├── GlobalOrthogonalRegularizer.py │ ├── L2Regularizer.py │ └── __init__.py ├── samplers │ ├── BaseSampler.py │ ├── TwoStageSampler.py │ └── __init__.py └── trainers │ ├── BaseTrainer.py │ └── __init__.py ├── README.md ├── README_ja.md ├── examples └── notebooks │ ├── movielens_cml.ipynb │ └── movielens_mf.ipynb ├── images ├── diagram.png └── icon.png ├── poetry.lock ├── pyproject.toml └── tests ├── __init__.py ├── adaptors ├── __init__.py └── test_MLPAdaptor.py ├── evaluators ├── __init__.py └── test_UserwiseEvaluator.py ├── losses ├── __init__.py ├── test_LogitPairwiseLoss.py ├── test_MSEPairwiseLoss.py ├── test_MinTripletLoss.py ├── test_RelevancePairwiseLoss.py └── test_SumTripletLoss.py ├── models ├── __init__.py ├── test_CollaborativeMetricLearning.py └── test_MatrixFactorization.py ├── regularizers ├── __init__.py ├── test_GlobalOrthogonalRegularizer.py └── test_L2Regularizer.py ├── samplers ├── __init__.py ├── test_BaseSampler.py └── test_TwoStageSampler.py └── trainers ├── __init__.py └── test_BaseTrainer.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.github/workflows/python-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/.github/workflows/python-test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/LICENSE -------------------------------------------------------------------------------- /PyTorchCML/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorchCML/adaptors/BaseAdaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/adaptors/BaseAdaptor.py -------------------------------------------------------------------------------- /PyTorchCML/adaptors/MLPAdaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/adaptors/MLPAdaptor.py -------------------------------------------------------------------------------- /PyTorchCML/adaptors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/adaptors/__init__.py -------------------------------------------------------------------------------- /PyTorchCML/evaluators/BaseEvaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/evaluators/BaseEvaluator.py -------------------------------------------------------------------------------- /PyTorchCML/evaluators/UserwiseEvaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/evaluators/UserwiseEvaluator.py -------------------------------------------------------------------------------- /PyTorchCML/evaluators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/evaluators/__init__.py -------------------------------------------------------------------------------- /PyTorchCML/losses/BaseLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/losses/BaseLoss.py -------------------------------------------------------------------------------- /PyTorchCML/losses/LogitPairwiseLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/losses/LogitPairwiseLoss.py -------------------------------------------------------------------------------- /PyTorchCML/losses/MSEPairwiseLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/losses/MSEPairwiseLoss.py -------------------------------------------------------------------------------- /PyTorchCML/losses/MinTripletLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/losses/MinTripletLoss.py -------------------------------------------------------------------------------- /PyTorchCML/losses/RelevancePairwiseLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/losses/RelevancePairwiseLoss.py -------------------------------------------------------------------------------- /PyTorchCML/losses/SumTripletLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/losses/SumTripletLoss.py -------------------------------------------------------------------------------- /PyTorchCML/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/losses/__init__.py -------------------------------------------------------------------------------- /PyTorchCML/models/BaseEmbeddingModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/models/BaseEmbeddingModel.py -------------------------------------------------------------------------------- /PyTorchCML/models/CollaborativeMetricLearning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/models/CollaborativeMetricLearning.py -------------------------------------------------------------------------------- /PyTorchCML/models/MatrixFactorization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/models/MatrixFactorization.py -------------------------------------------------------------------------------- /PyTorchCML/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/models/__init__.py -------------------------------------------------------------------------------- /PyTorchCML/regularizers/BaseRegularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/regularizers/BaseRegularizer.py -------------------------------------------------------------------------------- /PyTorchCML/regularizers/GlobalOrthogonalRegularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/regularizers/GlobalOrthogonalRegularizer.py -------------------------------------------------------------------------------- /PyTorchCML/regularizers/L2Regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/regularizers/L2Regularizer.py -------------------------------------------------------------------------------- /PyTorchCML/regularizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/regularizers/__init__.py -------------------------------------------------------------------------------- /PyTorchCML/samplers/BaseSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/samplers/BaseSampler.py -------------------------------------------------------------------------------- /PyTorchCML/samplers/TwoStageSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/samplers/TwoStageSampler.py -------------------------------------------------------------------------------- /PyTorchCML/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/samplers/__init__.py -------------------------------------------------------------------------------- /PyTorchCML/trainers/BaseTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/trainers/BaseTrainer.py -------------------------------------------------------------------------------- /PyTorchCML/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/PyTorchCML/trainers/__init__.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/README.md -------------------------------------------------------------------------------- /README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/README_ja.md -------------------------------------------------------------------------------- /examples/notebooks/movielens_cml.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/examples/notebooks/movielens_cml.ipynb -------------------------------------------------------------------------------- /examples/notebooks/movielens_mf.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/examples/notebooks/movielens_mf.ipynb -------------------------------------------------------------------------------- /images/diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/images/diagram.png -------------------------------------------------------------------------------- /images/icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/images/icon.png -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/adaptors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/adaptors/test_MLPAdaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/adaptors/test_MLPAdaptor.py -------------------------------------------------------------------------------- /tests/evaluators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evaluators/test_UserwiseEvaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/evaluators/test_UserwiseEvaluator.py -------------------------------------------------------------------------------- /tests/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/losses/test_LogitPairwiseLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/losses/test_LogitPairwiseLoss.py -------------------------------------------------------------------------------- /tests/losses/test_MSEPairwiseLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/losses/test_MSEPairwiseLoss.py -------------------------------------------------------------------------------- /tests/losses/test_MinTripletLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/losses/test_MinTripletLoss.py -------------------------------------------------------------------------------- /tests/losses/test_RelevancePairwiseLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/losses/test_RelevancePairwiseLoss.py -------------------------------------------------------------------------------- /tests/losses/test_SumTripletLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/losses/test_SumTripletLoss.py -------------------------------------------------------------------------------- /tests/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/models/test_CollaborativeMetricLearning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/models/test_CollaborativeMetricLearning.py -------------------------------------------------------------------------------- /tests/models/test_MatrixFactorization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/models/test_MatrixFactorization.py -------------------------------------------------------------------------------- /tests/regularizers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/regularizers/test_GlobalOrthogonalRegularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/regularizers/test_GlobalOrthogonalRegularizer.py -------------------------------------------------------------------------------- /tests/regularizers/test_L2Regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/regularizers/test_L2Regularizer.py -------------------------------------------------------------------------------- /tests/samplers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/samplers/test_BaseSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/samplers/test_BaseSampler.py -------------------------------------------------------------------------------- /tests/samplers/test_TwoStageSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/samplers/test_TwoStageSampler.py -------------------------------------------------------------------------------- /tests/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainers/test_BaseTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hand10ryo/PyTorchCML/HEAD/tests/trainers/test_BaseTrainer.py --------------------------------------------------------------------------------