diff --git a/README.md b/README.md index fab641c313a052663c34783f12c3b9ed8336e7cf..e29c13f0032c02f487b3430595358048912aa65d 100644 --- a/README.md +++ b/README.md @@ -107,7 +107,7 @@ The recommender models supported by Cornac are listed below. Why don't you join | Year | Model and paper | Additional dependencies | Examples | | :---: | --- | :---: | :---: | | 2021 | [Bilateral Variational Autoencoder for Collaborative Filtering (BiVAECF)](cornac/models/bivaecf), [paper](https://dl.acm.org/doi/pdf/10.1145/3437963.3441759) | [requirements.txt](cornac/models/bivaecf/requirements.txt) | [PreferredAI/bi-vae](https://github.com/PreferredAI/bi-vae) -| | [CausalRec: Causal Inference for Visual Debiasing in Visually-Aware Recommendation](cornac/models/causalrec), [paper](https://arxiv.org/abs/2107.02390) | N/A | [causalrec_clothing.py](examples/causalrec_clothing.py) +| | [Causal Inference for Visual Debiasing in Visually-Aware Recommendation (CausalRec)](cornac/models/causalrec), [paper](https://arxiv.org/abs/2107.02390) | [requirements.txt](cornac/models/causalrec/requirements.txt) | [causalrec_clothing.py](examples/causalrec_clothing.py) | | [Explainable Recommendation with Comparative Constraints on Product Aspects (ComparER)](cornac/models/comparer), [paper](https://dl.acm.org/doi/pdf/10.1145/3437963.3441754) | N/A | [PreferredAI/ComparER](https://github.com/PreferredAI/ComparER) | 2018 | [Collaborative Context Poisson Factorization (C2PF)](cornac/models/c2pf), [paper](https://www.ijcai.org/proceedings/2018/0370.pdf) | N/A | [c2pf_exp.py](examples/c2pf_example.py) | | [Multi-Task Explainable Recommendation (MTER)](cornac/models/mter), [paper](https://arxiv.org/pdf/1806.03568.pdf) | N/A | [mter_exp.py](examples/mter_example.py) diff --git a/cornac/models/causalrec/requirements.txt b/cornac/models/causalrec/requirements.txt new file mode 100644 index 0000000000000000000000000000000000000000..b54f97d6fe7bac67f432ccacbcba8b63a40add69 --- /dev/null +++ b/cornac/models/causalrec/requirements.txt @@ -0,0 +1 @@ +torch>=0.4.1 diff --git a/docs/source/models.rst b/docs/source/models.rst index 7ab7a74f6d91b6b82a0119b895bc88b4c7b10772..b4d39a2c9b2c314ca442c1b746dcba62e7e8b4c8 100644 --- a/docs/source/models.rst +++ b/docs/source/models.rst @@ -15,8 +15,8 @@ Bilateral VAE for Collaborative Filtering (BiVAECF) .. automodule:: cornac.models.bivaecf.recom_bivaecf :members: -CausalRec: Causal Inference for Visual Debiasing in Visually-Aware Recommendation (CausalRec) ---------------------------------------------------------------------------------------------- +Causal Inference for Visual Debiasing in Visually-Aware Recommendation (CausalRec) +---------------------------------------------------------------------------------- .. automodule:: cornac.models.causalrec.recom_causalrec :members: