Zhangya

docs/source/common.rst

@@ -18,6 +18,13 @@ GPU utilities
     :members:
 
 
+Kubernetes utilities
+===============================
+
+.. automodule:: reco_utils.common.k8s_utils
+    :members:
+
+
 Notebook utilities
 ===============================
 

docs/source/dataset.rst

@@ -12,6 +12,14 @@ Recommendation datasets
 .. automodule:: reco_utils.dataset.criteo
     :members:
 
+.. automodule:: reco_utils.dataset.amazon_reviews
+    :members:
+
+.. automodule:: reco_utils.dataset.covid_utils
+    :members:
+
+.. automodule:: reco_utils.dataset.mind
+    :members:  
 
 Download utilities 
 ===============================

docs/source/index.rst

@@ -11,7 +11,6 @@ evaluating recommender systems.
    :maxdepth: 1
    :caption: Contents:
 
-    AzureML <azureml>
     Common <common>
     Dataset <dataset>
     Evaluation <evaluation>

docs/source/recommender.rst

@@ -20,6 +20,9 @@ DeepRec
 .. automodule:: reco_utils.recommender.deeprec.models.dkn
     :members:
 
+.. automodule:: reco_utils.recommender.deeprec.models.dkn_item2item
+    :members:
+
 .. automodule:: reco_utils.recommender.deeprec.models.xDeepFM
     :members:
 
@@ -29,6 +32,18 @@ DeepRec
 .. automodule:: reco_utils.recommender.deeprec.io.dkn_iterator
     :members:
 
+.. automodule:: reco_utils.recommender.deeprec.io.dkn_item2item_iterator
+    :members:
+
+.. automodule:: reco_utils.recommender.deeprec.io.nextitnet_iterator
+    :members:
+
+.. automodule:: reco_utils.recommender.deeprec.io.sequential_iterator
+    :members:
+
+.. automodule:: reco_utils.recommender.deeprec.DataModel.ImplicitCF
+    :members:
+
 .. automodule:: reco_utils.recommender.deeprec.deeprec_utils
     :members:
 
@@ -38,6 +53,27 @@ FastAI
 .. automodule:: reco_utils.recommender.fastai.fastai_utils
     :members:
 
+Geoimc
+===============================
+
+.. automodule:: reco_utils.recommender.geoimc.geoimc_algorithm
+    :members:
+
+.. automodule:: reco_utils.recommender.geoimc.geoimc_data
+    :members:
+
+.. automodule:: reco_utils.recommender.geoimc.geoimc_predict
+    :members:
+
+.. automodule:: reco_utils.recommender.geoimc.geoimc_utils
+    :members:
+
+LightFM
+===============================
+
+.. automodule:: reco_utils.recommender.lightfm.lightfm_utils
+    :members:
+
 LightGBM
 ===============================
 
@@ -53,6 +89,35 @@ NCF
 .. automodule:: reco_utils.recommender.ncf.ncf_singlenode
     :members:
 
+Newsrec
+===============================
+
+.. automodule:: reco_utils.recommender.newsrec.io.mind_all_iterator
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.io.mind_iterator
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.models.base_model
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.models.layers
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.models.lstur
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.models.naml
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.models.npa
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.models.nrms
+    :members:
+
+.. automodule:: reco_utils.recommender.newsrec.newsrec_utils
+    :members:
 
 RBM
 ===============================
@@ -80,19 +145,31 @@ SAR
 .. automodule:: reco_utils.recommender.sar.sar_singlenode
     :members:
 
+
 Surprise
 ===============================
 
 .. automodule:: reco_utils.recommender.surprise.surprise_utils
     :members:
 
 
-Vowpal Wabbit
+TF-IDF
 ===============================
 
-.. automodule:: reco_utils.recommender.vowpal_wabbit.vw
+.. automodule:: reco_utils.recommender.tfidf.tfidf_utils
     :members:
 
+
+VAE
+===============================
+
+.. automodule:: reco_utils.recommender.vae.multinomial_vae
+    :members:
+
+.. automodule:: reco_utils.recommender.vae.standard_vae
+    :members:
+
+
 Wide & Deep
 ===============================
 

docs/source/tuning.rst

@@ -4,12 +4,6 @@ Hyperparameter tuning module
 *********************************
 Hyperparameter tuning module from Recommenders utilities.
 
-NNI utils
-===============================
-
-.. automodule:: reco_utils.tuning.nni.nni_utils
-    :members:
-
 Parameter sweep utils
 ===============================
 

reco_utils/tuning/parameter_sweep.py

@@ -8,17 +8,13 @@
 
 def generate_param_grid(params):
     """Generator of parameter grids.
-
     Generate parameter lists from a parameter dictionary in the form of:
-
     .. code-block:: python
         {
             "param1": [value1, value2],
             "param2": [value1, value2]
         }
-
     to:
-
     .. code-block:: python
         [
             {"param1": value1, "param2": value1},