CoRML - Colab

Joinn99 · Apr 18, 2023 · 9d51e5b · 9d51e5b
1 parent 57c3555
commit 9d51e5b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/CoRML Demonstration.ipynb b/CoRML Demonstration.ipynb
@@ -177,6 +177,10 @@
       },
       "outputs": [],
       "source": [
+        "#@title CoRML\n",
+        "#@markdown This cell includes a complete implementation of FPSR model.\n",
+        "#@markdown You can find it at [Github](https://github.com/Joinn99/CoRML/blob/torch/CoRML/model.py).\n",
+        "\n",
         "r\"\"\"\n",
         "CoRML (PyTorch Version)\n",
         "################################################\n",
@@ -527,6 +531,10 @@
         }
       ],
       "source": [
+        "#@title Running a test\n",
+        "#@markdown This cell will run a test on the specified dataset with the model CoRML.\n",
+        "#@markdown Reference [RecBole Doc](https://recbole.io/docs/developer_guide/customize_models.html).\n",
+        "\n",
         "r\"\"\"\n",
         "Code reference: https://recbole.io/docs/developer_guide/customize_models.html\n",
         "\"\"\"\n",