From 83f0fd0c1cb923f01651a2df56f71f817e8db66b Mon Sep 17 00:00:00 2001
From: Quoc-Tuan Truong <tqtg@users.noreply.github.com>
Date: Thu, 8 Jul 2021 17:42:57 +0800
Subject: [PATCH] Add requirements for CausalRec (#417)

---
 README.md                                | 2 +-
 cornac/models/causalrec/requirements.txt | 1 +
 docs/source/models.rst                   | 4 ++--
 3 files changed, 4 insertions(+), 3 deletions(-)
 create mode 100644 cornac/models/causalrec/requirements.txt

diff --git a/README.md b/README.md
index fab641c3..e29c13f0 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 00000000..b54f97d6
--- /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 7ab7a74f..b4d39a2c 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:
 
-- 
GitLab