Updated sasrec tutorial.

MobileTeleSystems · Oct 1, 2024 · aedf89d · aedf89d
1 parent b9526b2
commit aedf89d
Show file tree

Hide file tree

Showing 2 changed files with 374 additions and 195 deletions.
diff --git a/examples/sasrec_metrics_comp.ipynb b/examples/sasrec_metrics_comp.ipynb
@@ -1280,17 +1280,17 @@
     "RANDOM_STATE = 32\n",
     "ITERATIONS = 10\n",
     "\n",
-    "def make_base_model(factors: int, regularization: float, alpha: float, fit_features_together: bool=False):\n",
+    "def make_base_model(factors: int, regularization: float, alpha: float, fit_features_together: bool = False):\n",
     "    return ImplicitALSWrapperModel(\n",
     "        AlternatingLeastSquares(\n",
     "            factors=factors,\n",
     "            regularization=regularization,\n",
     "            alpha=alpha,\n",
     "            random_state=RANDOM_STATE,\n",
     "            use_gpu=False,\n",
-    "            num_threads = NUM_THREADS,\n",
+    "            num_threads=NUM_THREADS,\n",
     "            iterations=ITERATIONS),\n",
-    "        fit_features_together = fit_features_together,\n",
+    "        fit_features_together=fit_features_together,\n",
     "        )"
    ]
   },
@@ -1335,7 +1335,7 @@
     }
    ],
    "source": [
-    "model = make_base_model(factors = n_factors, regularization=regularization, alpha=alpha, fit_features_together=True)\n",
+    "model = make_base_model(factors=n_factors, regularization=regularization, alpha=alpha, fit_features_together=True)\n",
     "model.fit(dataset_full_features)\n",
     "recos = model.recommend(\n",
     "    users=test_users.astype(int),\n",
@@ -1513,13 +1513,6 @@
     "features_df"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
   {
    "cell_type": "code",
    "execution_count": null,