From 6904ac9e742de70bb8f92b50ab953499f9b7cb13 Mon Sep 17 00:00:00 2001
From: emustafa96 <emustafa@student.ethz.ch>
Date: Sun, 29 Oct 2023 13:45:19 +0100
Subject: [PATCH] update schema to generate emotions

---
 recommander/data_gen.ipynb | 12 ++++++------
 recommander/schema.json    | 21 +++++++++++++++++++++
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/recommander/data_gen.ipynb b/recommander/data_gen.ipynb
index c1bd4bb..615930f 100644
--- a/recommander/data_gen.ipynb
+++ b/recommander/data_gen.ipynb
@@ -18,7 +18,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 32,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -30,7 +30,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 33,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -44,18 +44,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 27,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
     "mockaroo_url = '\thttps://api.mockaroo.com/api/generate.csv'\n",
     "# API key\n",
-    "num_rows_to_generate = 10  # Max 1000 for the free version of the API"
+    "num_rows_to_generate = 20  # Max 1000 for the free version of the API"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 28,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -77,7 +77,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
diff --git a/recommander/schema.json b/recommander/schema.json
index 36fc399..0d9c388 100644
--- a/recommander/schema.json
+++ b/recommander/schema.json
@@ -105,5 +105,26 @@
         "min": 0,
         "max": 5,
         "decimals": 0
+    },
+    {
+        "name": "mood",
+        "type": "Number",
+        "min": 0,
+        "max": 5,
+        "decimals": 0
+    },
+    {
+        "name": "battery",
+        "type": "Number",
+        "min": 0,
+        "max": 5,
+        "decimals": 0
+    },
+    {
+        "name": "stress",
+        "type": "Number",
+        "min": 0,
+        "max": 5,
+        "decimals": 0
     }
 ]
\ No newline at end of file