final commit

notebooks/naive.ipynb

@@ -178,7 +178,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 3,
+      "execution_count": 44,
       "metadata": {
         "id": "PngL0QHq516u"
       },
@@ -202,7 +202,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 4,
+      "execution_count": 45,
       "metadata": {
         "id": "peZyue6t57Mz"
       },
@@ -217,7 +217,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 5,
+      "execution_count": 46,
       "metadata": {
         "id": "p68WNyaf58rS"
       },
@@ -230,7 +230,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 6,
+      "execution_count": 47,
       "metadata": {
         "id": "aSBKxRFa5-O_"
       },
@@ -244,7 +244,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 7,
+      "execution_count": 48,
       "metadata": {},
       "outputs": [],
       "source": [
@@ -256,7 +256,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 8,
+      "execution_count": 49,
       "metadata": {
         "id": "V1bhU5rW6BSY"
       },
@@ -271,7 +271,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 9,
+      "execution_count": 50,
       "metadata": {
         "id": "l6sUWKYC6DCw"
       },
@@ -282,7 +282,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 10,
+      "execution_count": 51,
       "metadata": {
         "id": "XxC0WnlL6EWz"
       },
@@ -295,7 +295,25 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 16,
+      "execution_count": 57,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "temp = df"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 78,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "df = temp.iloc[:100000]"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 79,
       "metadata": {
         "id": "5HLSc9z36Izn"
       },
@@ -310,7 +328,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 17,
+      "execution_count": 80,
       "metadata": {},
       "outputs": [],
       "source": [
@@ -319,7 +337,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 30,
+      "execution_count": 81,
       "metadata": {
         "id": "k47MaxR65Nq4"
       },
@@ -341,7 +359,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 36,
+      "execution_count": 82,
       "metadata": {},
       "outputs": [],
       "source": [
@@ -356,7 +374,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 38,
+      "execution_count": 83,
       "metadata": {
         "colab": {
           "base_uri": "https://localhost:8080/"
@@ -373,7 +391,24 @@
         "\n",
         "y_test_binary = (y_val >= 0.5).astype(int)\n",
         "y_test_items = X_val['album_id'][y_test_binary == 1].tolist()\n",
-        "\n",
+        "\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 84,
+      "metadata": {},
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "Precision: 1.0\n",
+            "Recall: 0.0011200716845879738\n"
+          ]
+        }
+      ],
+      "source": [
         "precisions = []\n",
         "recalls = []\n",
         "for i in range(len(X_val)):\n",