refactor: round metrics better

Ab2nour · Jan 15, 2024 · 88e8218 · 88e8218
1 parent 4c94631
commit 88e8218
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/notebooks/supervized_regression.ipynb b/src/notebooks/supervized_regression.ipynb
@@ -614,15 +614,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 34,
+   "execution_count": 35,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Random Forest Mean Squared Error = 229.26\n",
-      "Random Forest Mean Absolute Error = 11.66\n"
+      "Random Forest Mean Squared Error = 229\n",
+      "Random Forest Mean Absolute Error = 12\n"
      ]
     }
    ],
@@ -634,8 +634,8 @@
     "rf_mse = mean_squared_error(y_true=y_test, y_pred=y_pred_rf)\n",
     "rf_mae = mean_absolute_error(y_true=y_test, y_pred=y_pred_rf)\n",
     "\n",
-    "print(f\"Random Forest Mean Squared Error = {round(rf_mse, 2)}\")\n",
-    "print(f\"Random Forest Mean Absolute Error = {round(rf_mae, 2)}\")\n"
+    "print(f\"Random Forest Mean Squared Error = {round(rf_mse)}\")\n",
+    "print(f\"Random Forest Mean Absolute Error = {round(rf_mae)}\")\n"
    ]
   },
   {