From 5a6a48e6222e750c7ed06a262647d77b49cf4f4d Mon Sep 17 00:00:00 2001
From: Zakharov_Rostislav <r.z.8102.80@gmail.com>
Date: Sat, 14 Dec 2024 10:14:47 +0400
Subject: [PATCH] feat(lab-4): add prediction output

---
 notebooks/lab4.ipynb | 965 +++++++++++++++++++++++++++++++++----------
 1 file changed, 744 insertions(+), 221 deletions(-)

diff --git a/notebooks/lab4.ipynb b/notebooks/lab4.ipynb
index 5895840..d1ad90e 100644
--- a/notebooks/lab4.ipynb
+++ b/notebooks/lab4.ipynb
@@ -13,7 +13,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 732,
+   "execution_count": 30,
    "metadata": {},
    "outputs": [
     {
@@ -325,7 +325,7 @@
        "[19237 rows x 17 columns]"
       ]
      },
-     "execution_count": 732,
+     "execution_count": 30,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -375,7 +375,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 733,
+   "execution_count": 31,
    "metadata": {},
    "outputs": [
     {
@@ -401,7 +401,7 @@
        "dtype: object"
       ]
      },
-     "execution_count": 733,
+     "execution_count": 31,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -412,7 +412,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 734,
+   "execution_count": 32,
    "metadata": {},
    "outputs": [
     {
@@ -426,7 +426,7 @@
        "        6.8,  4.5,  7.3,  0.1,  3.1,  6.4,  3.9,  0.9,  5.2,  5.8])"
       ]
      },
-     "execution_count": 734,
+     "execution_count": 32,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -439,7 +439,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 735,
+   "execution_count": 33,
    "metadata": {},
    "outputs": [
     {
@@ -448,7 +448,7 @@
        "array([186005, 192000, 200000, ..., 140607, 307325, 186923])"
       ]
      },
-     "execution_count": 735,
+     "execution_count": 33,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -461,7 +461,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 736,
+   "execution_count": 34,
    "metadata": {},
    "outputs": [
     {
@@ -532,7 +532,7 @@
        "        1901])"
       ]
      },
-     "execution_count": 736,
+     "execution_count": 34,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -545,7 +545,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 737,
+   "execution_count": 35,
    "metadata": {},
    "outputs": [
     {
@@ -554,7 +554,7 @@
        "array([ 6,  4,  8,  1, 12,  3,  2, 16,  5,  7,  9, 10, 14])"
       ]
      },
-     "execution_count": 737,
+     "execution_count": 35,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -566,7 +566,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 738,
+   "execution_count": 36,
    "metadata": {},
    "outputs": [
     {
@@ -575,7 +575,7 @@
        "array(['04-May', '02-Mar', '>5'], dtype=object)"
       ]
      },
-     "execution_count": 738,
+     "execution_count": 36,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -586,7 +586,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 739,
+   "execution_count": 37,
    "metadata": {},
    "outputs": [
     {
@@ -595,7 +595,7 @@
        "array(['Четырехдверный', 'Двухдверный', 'Многодверный'], dtype=object)"
       ]
      },
-     "execution_count": 739,
+     "execution_count": 37,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -609,7 +609,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 740,
+   "execution_count": 38,
    "metadata": {},
    "outputs": [
     {
@@ -618,7 +618,7 @@
        "array([       1,        3,        6, ...,   627220,   872946, 26307500])"
       ]
      },
-     "execution_count": 740,
+     "execution_count": 38,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -630,7 +630,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 741,
+   "execution_count": 39,
    "metadata": {},
    "outputs": [
     {
@@ -650,7 +650,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 742,
+   "execution_count": 40,
    "metadata": {},
    "outputs": [
     {
@@ -659,7 +659,7 @@
        "array([     500,      549,      600, ...,   627220,   872946, 26307500])"
       ]
      },
-     "execution_count": 742,
+     "execution_count": 40,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -671,7 +671,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 743,
+   "execution_count": 41,
    "metadata": {},
    "outputs": [
     {
@@ -684,7 +684,7 @@
        "       2014, 2015, 2016, 2017, 2018, 2019, 2020])"
       ]
      },
-     "execution_count": 743,
+     "execution_count": 41,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -696,7 +696,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 744,
+   "execution_count": 42,
    "metadata": {},
    "outputs": [
     {
@@ -1021,7 +1021,7 @@
        "[17574 rows x 17 columns]"
       ]
      },
-     "execution_count": 744,
+     "execution_count": 42,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1039,7 +1039,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 745,
+   "execution_count": 43,
    "metadata": {},
    "outputs": [
     {
@@ -1048,7 +1048,7 @@
        "np.int64(2773)"
       ]
      },
-     "execution_count": 745,
+     "execution_count": 43,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1059,14 +1059,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 746,
+   "execution_count": 44,
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "C:\\Users\\user\\AppData\\Local\\Temp\\ipykernel_30576\\1689817098.py:1: SettingWithCopyWarning: \n",
+      "C:\\Users\\user\\AppData\\Local\\Temp\\ipykernel_18064\\1689817098.py:1: SettingWithCopyWarning: \n",
       "A value is trying to be set on a copy of a slice from a DataFrame\n",
       "\n",
       "See the caveats in the documentation: https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n",
@@ -1081,7 +1081,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 747,
+   "execution_count": 45,
    "metadata": {},
    "outputs": [
     {
@@ -1107,7 +1107,7 @@
        "dtype: int64"
       ]
      },
-     "execution_count": 747,
+     "execution_count": 45,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1125,7 +1125,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 748,
+   "execution_count": 46,
    "metadata": {},
    "outputs": [
     {
@@ -1151,7 +1151,7 @@
        "dtype: object"
       ]
      },
-     "execution_count": 748,
+     "execution_count": 46,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1162,7 +1162,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 749,
+   "execution_count": 47,
    "metadata": {},
    "outputs": [
     {
@@ -1224,7 +1224,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 750,
+   "execution_count": 48,
    "metadata": {},
    "outputs": [
     {
@@ -1255,7 +1255,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 751,
+   "execution_count": 49,
    "metadata": {},
    "outputs": [
     {
@@ -1317,7 +1317,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 752,
+   "execution_count": 50,
    "metadata": {},
    "outputs": [
     {
@@ -1343,7 +1343,7 @@
        "dtype: object"
       ]
      },
-     "execution_count": 752,
+     "execution_count": 50,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1354,13 +1354,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 753,
+   "execution_count": 51,
    "metadata": {},
    "outputs": [],
    "source": [
     "columns_to_drop = [\n",
     "    \"Model\",\n",
-    "    \"Manufacturer\",\n",
+    "    # \"Manufacturer\",\n",
     "    \"Color\",\n",
     "    \"Doors\",\n",
     "    \"Cylinders\",\n",
@@ -1387,6 +1387,7 @@
     "    \"Drive wheels\",\n",
     "    # \"Doors\",\n",
     "    \"Wheel\",\n",
+    "    \"Manufacturer\",\n",
     "]\n",
     "\n",
     "cat_cols_for_num_enc = [\n",
@@ -1457,7 +1458,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 754,
+   "execution_count": 52,
    "metadata": {},
    "outputs": [
     {
@@ -1492,11 +1493,11 @@
        "      <th>Category_Pickup</th>\n",
        "      <th>Category_Sedan</th>\n",
        "      <th>...</th>\n",
-       "      <th>Gear box type_Tiptronic</th>\n",
-       "      <th>Gear box type_Variator</th>\n",
-       "      <th>Drive wheels_Front</th>\n",
-       "      <th>Drive wheels_Rear</th>\n",
-       "      <th>Wheel_Right-hand drive</th>\n",
+       "      <th>Manufacturer_UAZ</th>\n",
+       "      <th>Manufacturer_VAZ</th>\n",
+       "      <th>Manufacturer_VOLKSWAGEN</th>\n",
+       "      <th>Manufacturer_VOLVO</th>\n",
+       "      <th>Manufacturer_სხვა</th>\n",
        "      <th>Price</th>\n",
        "      <th>Levy</th>\n",
        "      <th>Prod. year</th>\n",
@@ -1542,7 +1543,7 @@
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>...</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
@@ -1567,10 +1568,10 @@
        "      <td>0.0</td>\n",
        "      <td>...</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>-0.781060</td>\n",
        "      <td>-1.187596</td>\n",
        "      <td>-1.091356</td>\n",
@@ -1616,7 +1617,7 @@
        "      <td>...</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>-0.480485</td>\n",
@@ -1664,7 +1665,7 @@
        "      <td>...</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>-0.752100</td>\n",
@@ -1686,11 +1687,11 @@
        "      <td>0.0</td>\n",
        "      <td>1.0</td>\n",
        "      <td>...</td>\n",
-       "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>-0.838888</td>\n",
        "      <td>-1.187596</td>\n",
        "      <td>-1.741709</td>\n",
@@ -1713,7 +1714,7 @@
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>-0.781060</td>\n",
        "      <td>-1.187596</td>\n",
@@ -1734,9 +1735,9 @@
        "      <td>0.0</td>\n",
        "      <td>1.0</td>\n",
        "      <td>...</td>\n",
-       "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>-0.115718</td>\n",
@@ -1760,7 +1761,7 @@
        "      <td>...</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.845956</td>\n",
@@ -1771,7 +1772,7 @@
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n",
-       "<p>12597 rows × 28 columns</p>\n",
+       "<p>12597 rows × 83 columns</p>\n",
        "</div>"
       ],
       "text/plain": [
@@ -1814,49 +1815,49 @@
        "19233                0.0               0.0              0.0             1.0   \n",
        "19234                0.0               0.0              0.0             0.0   \n",
        "\n",
-       "       ...  Gear box type_Tiptronic  Gear box type_Variator  \\\n",
-       "0      ...                      0.0                     0.0   \n",
-       "1      ...                      1.0                     0.0   \n",
-       "2      ...                      0.0                     1.0   \n",
-       "3      ...                      0.0                     0.0   \n",
-       "4      ...                      0.0                     0.0   \n",
-       "...    ...                      ...                     ...   \n",
-       "19225  ...                      0.0                     0.0   \n",
-       "19226  ...                      1.0                     0.0   \n",
-       "19232  ...                      0.0                     0.0   \n",
-       "19233  ...                      1.0                     0.0   \n",
-       "19234  ...                      0.0                     0.0   \n",
+       "       ...  Manufacturer_UAZ  Manufacturer_VAZ  Manufacturer_VOLKSWAGEN  \\\n",
+       "0      ...               0.0               0.0                      0.0   \n",
+       "1      ...               0.0               0.0                      0.0   \n",
+       "2      ...               0.0               0.0                      0.0   \n",
+       "3      ...               0.0               0.0                      0.0   \n",
+       "4      ...               0.0               0.0                      0.0   \n",
+       "...    ...               ...               ...                      ...   \n",
+       "19225  ...               0.0               0.0                      0.0   \n",
+       "19226  ...               0.0               0.0                      0.0   \n",
+       "19232  ...               0.0               0.0                      0.0   \n",
+       "19233  ...               0.0               0.0                      0.0   \n",
+       "19234  ...               0.0               0.0                      0.0   \n",
        "\n",
-       "       Drive wheels_Front  Drive wheels_Rear  Wheel_Right-hand drive  \\\n",
-       "0                     0.0                0.0                     0.0   \n",
-       "1                     0.0                0.0                     0.0   \n",
-       "2                     1.0                0.0                     1.0   \n",
-       "3                     0.0                0.0                     0.0   \n",
-       "4                     1.0                0.0                     0.0   \n",
-       "...                   ...                ...                     ...   \n",
-       "19225                 1.0                0.0                     0.0   \n",
-       "19226                 0.0                1.0                     1.0   \n",
-       "19232                 0.0                1.0                     0.0   \n",
-       "19233                 1.0                0.0                     0.0   \n",
-       "19234                 1.0                0.0                     0.0   \n",
+       "       Manufacturer_VOLVO  Manufacturer_სხვა     Price      Levy  Prod. year  \\\n",
+       "0                     0.0                0.0 -0.332733  1.851925   -0.224219   \n",
+       "1                     0.0                0.0 -0.029023  1.024150   -0.007434   \n",
+       "2                     0.0                0.0 -0.781060 -1.187596   -1.091356   \n",
+       "3                     0.0                0.0 -1.229294  0.685218   -0.007434   \n",
+       "4                     0.0                0.0 -0.480485 -0.218599    0.642919   \n",
+       "...                   ...                ...       ...       ...         ...   \n",
+       "19225                 0.0                0.0 -0.752100  1.217515   -0.874572   \n",
+       "19226                 0.0                0.0 -0.838888 -1.187596   -1.741709   \n",
+       "19232                 0.0                0.0 -0.781060 -1.187596   -2.608847   \n",
+       "19233                 0.0                0.0 -0.115718  0.617867   -0.007434   \n",
+       "19234                 0.0                0.0  0.845956  0.628730   -0.224219   \n",
        "\n",
-       "          Price      Levy  Prod. year  Engine volume   Airbags  \n",
-       "0     -0.332733  1.851925   -0.224219       1.849443  1.353782  \n",
-       "1     -0.029023  1.024150   -0.007434       1.162057  0.340316  \n",
-       "2     -0.781060 -1.187596   -1.091356      -1.175055 -1.179883  \n",
-       "3     -1.229294  0.685218   -0.007434       0.474671 -1.686616  \n",
-       "4     -0.480485 -0.218599    0.642919      -1.175055 -0.673150  \n",
-       "...         ...       ...         ...            ...       ...  \n",
-       "19225 -0.752100  1.217515   -0.874572      -0.625146 -0.673150  \n",
-       "19226 -0.838888 -1.187596   -1.741709       1.162057 -1.686616  \n",
-       "19232 -0.781060 -1.187596   -2.608847      -0.212715 -0.419784  \n",
-       "19233 -0.115718  0.617867   -0.007434       0.337194  0.340316  \n",
-       "19234  0.845956  0.628730   -0.224219      -0.212715 -0.673150  \n",
+       "       Engine volume   Airbags  \n",
+       "0           1.849443  1.353782  \n",
+       "1           1.162057  0.340316  \n",
+       "2          -1.175055 -1.179883  \n",
+       "3           0.474671 -1.686616  \n",
+       "4          -1.175055 -0.673150  \n",
+       "...              ...       ...  \n",
+       "19225      -0.625146 -0.673150  \n",
+       "19226       1.162057 -1.686616  \n",
+       "19232      -0.212715 -0.419784  \n",
+       "19233       0.337194  0.340316  \n",
+       "19234      -0.212715 -0.673150  \n",
        "\n",
-       "[12597 rows x 28 columns]"
+       "[12597 rows x 83 columns]"
       ]
      },
-     "execution_count": 754,
+     "execution_count": 52,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1880,7 +1881,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 755,
+   "execution_count": 53,
    "metadata": {},
    "outputs": [
     {
@@ -1905,7 +1906,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 756,
+   "execution_count": 54,
    "metadata": {},
    "outputs": [
     {
@@ -1940,12 +1941,12 @@
        "      <th>Category_Pickup</th>\n",
        "      <th>Category_Sedan</th>\n",
        "      <th>...</th>\n",
-       "      <th>Gear box type_Manual</th>\n",
-       "      <th>Gear box type_Tiptronic</th>\n",
-       "      <th>Gear box type_Variator</th>\n",
-       "      <th>Drive wheels_Front</th>\n",
-       "      <th>Drive wheels_Rear</th>\n",
-       "      <th>Wheel_Right-hand drive</th>\n",
+       "      <th>Manufacturer_TOYOTA</th>\n",
+       "      <th>Manufacturer_UAZ</th>\n",
+       "      <th>Manufacturer_VAZ</th>\n",
+       "      <th>Manufacturer_VOLKSWAGEN</th>\n",
+       "      <th>Manufacturer_VOLVO</th>\n",
+       "      <th>Manufacturer_სხვა</th>\n",
        "      <th>Levy</th>\n",
        "      <th>Prod. year</th>\n",
        "      <th>Engine volume</th>\n",
@@ -1966,12 +1967,12 @@
        "      <td>0.0</td>\n",
        "      <td>1.0</td>\n",
        "      <td>...</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
-       "      <td>0.0</td>\n",
        "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>-1.187596</td>\n",
        "      <td>1.943625</td>\n",
        "      <td>-0.487669</td>\n",
@@ -1990,12 +1991,12 @@
        "      <td>0.0</td>\n",
        "      <td>1.0</td>\n",
        "      <td>...</td>\n",
-       "      <td>0.0</td>\n",
        "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>-1.187596</td>\n",
        "      <td>-0.874572</td>\n",
        "      <td>-0.900100</td>\n",
@@ -2015,9 +2016,9 @@
        "      <td>1.0</td>\n",
        "      <td>...</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.081225</td>\n",
@@ -2065,7 +2066,7 @@
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>1.695495</td>\n",
@@ -2110,7 +2111,7 @@
        "      <td>0.0</td>\n",
        "      <td>1.0</td>\n",
        "      <td>...</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
@@ -2137,7 +2138,7 @@
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.474472</td>\n",
@@ -2158,10 +2159,10 @@
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>...</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
-       "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.643938</td>\n",
@@ -2185,7 +2186,7 @@
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>-1.187596</td>\n",
@@ -2207,9 +2208,9 @@
        "      <td>1.0</td>\n",
        "      <td>...</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
        "      <td>0.0</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.0</td>\n",
        "      <td>0.444055</td>\n",
@@ -2219,7 +2220,7 @@
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n",
-       "<p>8817 rows × 27 columns</p>\n",
+       "<p>8817 rows × 82 columns</p>\n",
        "</div>"
       ],
       "text/plain": [
@@ -2262,49 +2263,49 @@
        "1136                 0.0               0.0              0.0             1.0   \n",
        "10640                0.0               0.0              0.0             1.0   \n",
        "\n",
-       "       ...  Gear box type_Manual  Gear box type_Tiptronic  \\\n",
-       "15146  ...                   0.0                      0.0   \n",
-       "14145  ...                   0.0                      1.0   \n",
-       "8943   ...                   0.0                      1.0   \n",
-       "17889  ...                   0.0                      0.0   \n",
-       "9515   ...                   0.0                      0.0   \n",
-       "...    ...                   ...                      ...   \n",
-       "18201  ...                   1.0                      0.0   \n",
-       "7436   ...                   0.0                      0.0   \n",
-       "7728   ...                   0.0                      0.0   \n",
-       "1136   ...                   0.0                      0.0   \n",
-       "10640  ...                   0.0                      1.0   \n",
+       "       ...  Manufacturer_TOYOTA  Manufacturer_UAZ  Manufacturer_VAZ  \\\n",
+       "15146  ...                  1.0               0.0               0.0   \n",
+       "14145  ...                  1.0               0.0               0.0   \n",
+       "8943   ...                  0.0               0.0               0.0   \n",
+       "17889  ...                  0.0               0.0               0.0   \n",
+       "9515   ...                  0.0               0.0               0.0   \n",
+       "...    ...                  ...               ...               ...   \n",
+       "18201  ...                  0.0               0.0               0.0   \n",
+       "7436   ...                  0.0               0.0               0.0   \n",
+       "7728   ...                  1.0               0.0               0.0   \n",
+       "1136   ...                  0.0               0.0               0.0   \n",
+       "10640  ...                  0.0               0.0               0.0   \n",
        "\n",
-       "       Gear box type_Variator  Drive wheels_Front  Drive wheels_Rear  \\\n",
-       "15146                     0.0                 1.0                0.0   \n",
-       "14145                     0.0                 1.0                0.0   \n",
-       "8943                      0.0                 1.0                0.0   \n",
-       "17889                     0.0                 0.0                0.0   \n",
-       "9515                      0.0                 1.0                0.0   \n",
-       "...                       ...                 ...                ...   \n",
-       "18201                     0.0                 0.0                0.0   \n",
-       "7436                      0.0                 1.0                0.0   \n",
-       "7728                      1.0                 1.0                0.0   \n",
-       "1136                      0.0                 1.0                0.0   \n",
-       "10640                     0.0                 1.0                0.0   \n",
+       "       Manufacturer_VOLKSWAGEN  Manufacturer_VOLVO  Manufacturer_სხვა  \\\n",
+       "15146                      0.0                 0.0                0.0   \n",
+       "14145                      0.0                 0.0                0.0   \n",
+       "8943                       0.0                 0.0                0.0   \n",
+       "17889                      0.0                 0.0                0.0   \n",
+       "9515                       0.0                 0.0                0.0   \n",
+       "...                        ...                 ...                ...   \n",
+       "18201                      0.0                 0.0                0.0   \n",
+       "7436                       0.0                 0.0                0.0   \n",
+       "7728                       0.0                 0.0                0.0   \n",
+       "1136                       0.0                 0.0                0.0   \n",
+       "10640                      0.0                 0.0                0.0   \n",
        "\n",
-       "       Wheel_Right-hand drive      Levy  Prod. year  Engine volume   Airbags  \n",
-       "15146                     0.0 -1.187596    1.943625      -0.487669 -0.673150  \n",
-       "14145                     1.0 -1.187596   -0.874572      -0.900100 -1.179883  \n",
-       "8943                      0.0  0.081225    0.642919      -0.487669  0.847049  \n",
-       "17889                     0.0 -1.187596   -1.524925       0.474671 -0.419784  \n",
-       "9515                      0.0  1.695495    1.510056       0.474671 -0.673150  \n",
-       "...                       ...       ...         ...            ...       ...  \n",
-       "18201                     0.0 -1.187596   -0.874572       0.337194 -0.673150  \n",
-       "7436                      0.0  0.474472    0.859703      -0.212715  1.353782  \n",
-       "7728                      0.0  0.643938   -0.657787      -0.900100  0.340316  \n",
-       "1136                      0.0 -1.187596    1.076487      -0.762623  0.340316  \n",
-       "10640                     0.0  0.444055    0.426134       0.337194  0.340316  \n",
+       "           Levy  Prod. year  Engine volume   Airbags  \n",
+       "15146 -1.187596    1.943625      -0.487669 -0.673150  \n",
+       "14145 -1.187596   -0.874572      -0.900100 -1.179883  \n",
+       "8943   0.081225    0.642919      -0.487669  0.847049  \n",
+       "17889 -1.187596   -1.524925       0.474671 -0.419784  \n",
+       "9515   1.695495    1.510056       0.474671 -0.673150  \n",
+       "...         ...         ...            ...       ...  \n",
+       "18201 -1.187596   -0.874572       0.337194 -0.673150  \n",
+       "7436   0.474472    0.859703      -0.212715  1.353782  \n",
+       "7728   0.643938   -0.657787      -0.900100  0.340316  \n",
+       "1136  -1.187596    1.076487      -0.762623  0.340316  \n",
+       "10640  0.444055    0.426134       0.337194  0.340316  \n",
        "\n",
-       "[8817 rows x 27 columns]"
+       "[8817 rows x 82 columns]"
       ]
      },
-     "execution_count": 756,
+     "execution_count": 54,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -2327,7 +2328,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 757,
+   "execution_count": 55,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -2367,17 +2368,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 758,
+   "execution_count": 56,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Model: linear\n",
-      "Model: ridge\n",
-      "Model: decision_tree\n",
-      "Model: knn\n"
+      "Model: linear\n"
      ]
     },
     {
@@ -2387,7 +2385,20 @@
       "c:\\Users\\user\\source\\repos\\mai_pi-33_zakharov\\.venv\\Lib\\site-packages\\sklearn\\base.py:486: UserWarning: X has feature names, but LinearRegression was fitted without feature names\n",
       "  warnings.warn(\n",
       "c:\\Users\\user\\source\\repos\\mai_pi-33_zakharov\\.venv\\Lib\\site-packages\\sklearn\\base.py:486: UserWarning: X has feature names, but LinearRegression was fitted without feature names\n",
-      "  warnings.warn(\n",
+      "  warnings.warn(\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Model: ridge\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
       "c:\\Users\\user\\source\\repos\\mai_pi-33_zakharov\\.venv\\Lib\\site-packages\\sklearn\\base.py:486: UserWarning: X has feature names, but RidgeCV was fitted without feature names\n",
       "  warnings.warn(\n",
       "c:\\Users\\user\\source\\repos\\mai_pi-33_zakharov\\.venv\\Lib\\site-packages\\sklearn\\base.py:486: UserWarning: X has feature names, but RidgeCV was fitted without feature names\n",
@@ -2406,6 +2417,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "Model: decision_tree\n",
+      "Model: knn\n",
       "Model: random_forest\n"
      ]
     },
@@ -2456,116 +2469,116 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 759,
+   "execution_count": 57,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/html": [
        "<style type=\"text/css\">\n",
-       "#T_6c98d_row0_col0, #T_6c98d_row0_col1 {\n",
+       "#T_d4e42_row0_col0, #T_d4e42_row0_col1 {\n",
        "  background-color: #26818e;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row0_col2, #T_6c98d_row3_col3, #T_6c98d_row4_col3 {\n",
+       "#T_d4e42_row0_col2, #T_d4e42_row3_col3, #T_d4e42_row4_col3 {\n",
        "  background-color: #4e02a2;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row0_col3, #T_6c98d_row3_col2, #T_6c98d_row4_col2 {\n",
+       "#T_d4e42_row0_col3, #T_d4e42_row3_col2, #T_d4e42_row4_col2 {\n",
        "  background-color: #da5a6a;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row1_col0 {\n",
-       "  background-color: #1fa188;\n",
+       "#T_d4e42_row1_col0 {\n",
+       "  background-color: #21a585;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row1_col1 {\n",
-       "  background-color: #20928c;\n",
+       "#T_d4e42_row1_col1 {\n",
+       "  background-color: #1f988b;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row1_col2 {\n",
-       "  background-color: #7801a8;\n",
+       "#T_d4e42_row1_col2 {\n",
+       "  background-color: #8305a7;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row1_col3 {\n",
-       "  background-color: #cb4679;\n",
+       "#T_d4e42_row1_col3 {\n",
+       "  background-color: #c5407e;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row2_col0 {\n",
-       "  background-color: #2cb17e;\n",
+       "#T_d4e42_row2_col0 {\n",
+       "  background-color: #34b679;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row2_col1 {\n",
-       "  background-color: #27ad81;\n",
+       "#T_d4e42_row2_col1 {\n",
+       "  background-color: #2eb37c;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row2_col2 {\n",
-       "  background-color: #9613a1;\n",
+       "#T_d4e42_row2_col2 {\n",
+       "  background-color: #a01a9c;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row2_col3 {\n",
-       "  background-color: #ad2793;\n",
+       "#T_d4e42_row2_col3 {\n",
+       "  background-color: #a72197;\n",
        "  color: #f1f1f1;\n",
        "}\n",
-       "#T_6c98d_row3_col0, #T_6c98d_row3_col1, #T_6c98d_row4_col0, #T_6c98d_row4_col1 {\n",
+       "#T_d4e42_row3_col0, #T_d4e42_row3_col1, #T_d4e42_row4_col0, #T_d4e42_row4_col1 {\n",
        "  background-color: #a8db34;\n",
        "  color: #000000;\n",
        "}\n",
        "</style>\n",
-       "<table id=\"T_6c98d\">\n",
+       "<table id=\"T_d4e42\">\n",
        "  <thead>\n",
        "    <tr>\n",
        "      <th class=\"blank level0\" >&nbsp;</th>\n",
-       "      <th id=\"T_6c98d_level0_col0\" class=\"col_heading level0 col0\" >RMSE_train</th>\n",
-       "      <th id=\"T_6c98d_level0_col1\" class=\"col_heading level0 col1\" >RMSE_test</th>\n",
-       "      <th id=\"T_6c98d_level0_col2\" class=\"col_heading level0 col2\" >RMAE_test</th>\n",
-       "      <th id=\"T_6c98d_level0_col3\" class=\"col_heading level0 col3\" >R2_test</th>\n",
+       "      <th id=\"T_d4e42_level0_col0\" class=\"col_heading level0 col0\" >RMSE_train</th>\n",
+       "      <th id=\"T_d4e42_level0_col1\" class=\"col_heading level0 col1\" >RMSE_test</th>\n",
+       "      <th id=\"T_d4e42_level0_col2\" class=\"col_heading level0 col2\" >RMAE_test</th>\n",
+       "      <th id=\"T_d4e42_level0_col3\" class=\"col_heading level0 col3\" >R2_test</th>\n",
        "    </tr>\n",
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <th id=\"T_6c98d_level0_row0\" class=\"row_heading level0 row0\" >knn</th>\n",
-       "      <td id=\"T_6c98d_row0_col0\" class=\"data row0 col0\" >0.536712</td>\n",
-       "      <td id=\"T_6c98d_row0_col1\" class=\"data row0 col1\" >0.602377</td>\n",
-       "      <td id=\"T_6c98d_row0_col2\" class=\"data row0 col2\" >0.643601</td>\n",
-       "      <td id=\"T_6c98d_row0_col3\" class=\"data row0 col3\" >0.640894</td>\n",
+       "      <th id=\"T_d4e42_level0_row0\" class=\"row_heading level0 row0\" >knn</th>\n",
+       "      <td id=\"T_d4e42_row0_col0\" class=\"data row0 col0\" >0.529359</td>\n",
+       "      <td id=\"T_d4e42_row0_col1\" class=\"data row0 col1\" >0.594354</td>\n",
+       "      <td id=\"T_d4e42_row0_col2\" class=\"data row0 col2\" >0.635191</td>\n",
+       "      <td id=\"T_d4e42_row0_col3\" class=\"data row0 col3\" >0.650397</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th id=\"T_6c98d_level0_row1\" class=\"row_heading level0 row1\" >random_forest</th>\n",
-       "      <td id=\"T_6c98d_row1_col0\" class=\"data row1 col0\" >0.610511</td>\n",
-       "      <td id=\"T_6c98d_row1_col1\" class=\"data row1 col1\" >0.634321</td>\n",
-       "      <td id=\"T_6c98d_row1_col2\" class=\"data row1 col2\" >0.675524</td>\n",
-       "      <td id=\"T_6c98d_row1_col3\" class=\"data row1 col3\" >0.601798</td>\n",
+       "      <th id=\"T_d4e42_level0_row1\" class=\"row_heading level0 row1\" >random_forest</th>\n",
+       "      <td id=\"T_d4e42_row1_col0\" class=\"data row1 col0\" >0.609210</td>\n",
+       "      <td id=\"T_d4e42_row1_col1\" class=\"data row1 col1\" >0.631618</td>\n",
+       "      <td id=\"T_d4e42_row1_col2\" class=\"data row1 col2\" >0.673173</td>\n",
+       "      <td id=\"T_d4e42_row1_col3\" class=\"data row1 col3\" >0.605184</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th id=\"T_6c98d_level0_row2\" class=\"row_heading level0 row2\" >decision_tree</th>\n",
-       "      <td id=\"T_6c98d_row2_col0\" class=\"data row2 col0\" >0.650704</td>\n",
-       "      <td id=\"T_6c98d_row2_col1\" class=\"data row2 col1\" >0.683437</td>\n",
-       "      <td id=\"T_6c98d_row2_col2\" class=\"data row2 col2\" >0.701393</td>\n",
-       "      <td id=\"T_6c98d_row2_col3\" class=\"data row2 col3\" >0.537743</td>\n",
+       "      <th id=\"T_d4e42_level0_row2\" class=\"row_heading level0 row2\" >decision_tree</th>\n",
+       "      <td id=\"T_d4e42_row2_col0\" class=\"data row2 col0\" >0.647025</td>\n",
+       "      <td id=\"T_d4e42_row2_col1\" class=\"data row2 col1\" >0.674358</td>\n",
+       "      <td id=\"T_d4e42_row2_col2\" class=\"data row2 col2\" >0.696539</td>\n",
+       "      <td id=\"T_d4e42_row2_col3\" class=\"data row2 col3\" >0.549944</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th id=\"T_6c98d_level0_row3\" class=\"row_heading level0 row3\" >ridge</th>\n",
-       "      <td id=\"T_6c98d_row3_col0\" class=\"data row3 col0\" >0.779199</td>\n",
-       "      <td id=\"T_6c98d_row3_col1\" class=\"data row3 col1\" >0.788052</td>\n",
-       "      <td id=\"T_6c98d_row3_col2\" class=\"data row3 col2\" >0.776391</td>\n",
-       "      <td id=\"T_6c98d_row3_col3\" class=\"data row3 col3\" >0.385397</td>\n",
+       "      <th id=\"T_d4e42_level0_row3\" class=\"row_heading level0 row3\" >ridge</th>\n",
+       "      <td id=\"T_d4e42_row3_col0\" class=\"data row3 col0\" >0.753637</td>\n",
+       "      <td id=\"T_d4e42_row3_col1\" class=\"data row3 col1\" >0.759154</td>\n",
+       "      <td id=\"T_d4e42_row3_col2\" class=\"data row3 col2\" >0.758869</td>\n",
+       "      <td id=\"T_d4e42_row3_col3\" class=\"data row3 col3\" >0.429644</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th id=\"T_6c98d_level0_row4\" class=\"row_heading level0 row4\" >linear</th>\n",
-       "      <td id=\"T_6c98d_row4_col0\" class=\"data row4 col0\" >0.779198</td>\n",
-       "      <td id=\"T_6c98d_row4_col1\" class=\"data row4 col1\" >0.788056</td>\n",
-       "      <td id=\"T_6c98d_row4_col2\" class=\"data row4 col2\" >0.776379</td>\n",
-       "      <td id=\"T_6c98d_row4_col3\" class=\"data row4 col3\" >0.385390</td>\n",
+       "      <th id=\"T_d4e42_level0_row4\" class=\"row_heading level0 row4\" >linear</th>\n",
+       "      <td id=\"T_d4e42_row4_col0\" class=\"data row4 col0\" >0.752569</td>\n",
+       "      <td id=\"T_d4e42_row4_col1\" class=\"data row4 col1\" >0.759341</td>\n",
+       "      <td id=\"T_d4e42_row4_col2\" class=\"data row4 col2\" >0.758784</td>\n",
+       "      <td id=\"T_d4e42_row4_col3\" class=\"data row4 col3\" >0.429364</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n"
       ],
       "text/plain": [
-       "<pandas.io.formats.style.Styler at 0x1a3418bf980>"
+       "<pandas.io.formats.style.Styler at 0x12df22077d0>"
       ]
      },
-     "execution_count": 759,
+     "execution_count": 57,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -2578,6 +2591,516 @@
     "    cmap=\"viridis\", low=1, high=0.3, subset=[\"RMSE_train\", \"RMSE_test\"]\n",
     ").background_gradient(cmap=\"plasma\", low=0.3, high=1, subset=[\"RMAE_test\", \"R2_test\"])"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 58,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'knn'"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "best_model = str(reg_metrics.sort_values(by=\"RMSE_test\").iloc[0].name)\n",
+    "\n",
+    "display(best_model)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "#### Вывод предсказаний для выборок"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 59,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>Leather interior_Yes</th>\n",
+       "      <th>Category_Coupe</th>\n",
+       "      <th>Category_Goods wagon</th>\n",
+       "      <th>Category_Hatchback</th>\n",
+       "      <th>Category_Jeep</th>\n",
+       "      <th>Category_Limousine</th>\n",
+       "      <th>Category_Microbus</th>\n",
+       "      <th>Category_Minivan</th>\n",
+       "      <th>Category_Pickup</th>\n",
+       "      <th>Category_Sedan</th>\n",
+       "      <th>...</th>\n",
+       "      <th>Manufacturer_VAZ</th>\n",
+       "      <th>Manufacturer_VOLKSWAGEN</th>\n",
+       "      <th>Manufacturer_VOLVO</th>\n",
+       "      <th>Manufacturer_სხვა</th>\n",
+       "      <th>Levy</th>\n",
+       "      <th>Prod. year</th>\n",
+       "      <th>Engine volume</th>\n",
+       "      <th>Airbags</th>\n",
+       "      <th>Price</th>\n",
+       "      <th>PricePred</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>15146</th>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>-1.187596</td>\n",
+       "      <td>1.943625</td>\n",
+       "      <td>-0.487669</td>\n",
+       "      <td>-0.673150</td>\n",
+       "      <td>0.144553</td>\n",
+       "      <td>0.557766</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>14145</th>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>-1.187596</td>\n",
+       "      <td>-0.874572</td>\n",
+       "      <td>-0.900100</td>\n",
+       "      <td>-1.179883</td>\n",
+       "      <td>-0.665312</td>\n",
+       "      <td>-0.603334</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>8943</th>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.081225</td>\n",
+       "      <td>0.642919</td>\n",
+       "      <td>-0.487669</td>\n",
+       "      <td>0.847049</td>\n",
+       "      <td>-0.144678</td>\n",
+       "      <td>-0.355422</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>17889</th>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>-1.187596</td>\n",
+       "      <td>-1.524925</td>\n",
+       "      <td>0.474671</td>\n",
+       "      <td>-0.419784</td>\n",
+       "      <td>-0.462869</td>\n",
+       "      <td>-0.746737</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>9515</th>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.695495</td>\n",
+       "      <td>1.510056</td>\n",
+       "      <td>0.474671</td>\n",
+       "      <td>-0.673150</td>\n",
+       "      <td>2.765339</td>\n",
+       "      <td>2.375842</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "<p>5 rows × 84 columns</p>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "       Leather interior_Yes  Category_Coupe  Category_Goods wagon  \\\n",
+       "15146                   0.0             0.0                   0.0   \n",
+       "14145                   0.0             0.0                   0.0   \n",
+       "8943                    0.0             0.0                   0.0   \n",
+       "17889                   1.0             0.0                   0.0   \n",
+       "9515                    1.0             0.0                   0.0   \n",
+       "\n",
+       "       Category_Hatchback  Category_Jeep  Category_Limousine  \\\n",
+       "15146                 0.0            0.0                 0.0   \n",
+       "14145                 0.0            0.0                 0.0   \n",
+       "8943                  0.0            0.0                 0.0   \n",
+       "17889                 0.0            1.0                 0.0   \n",
+       "9515                  0.0            0.0                 0.0   \n",
+       "\n",
+       "       Category_Microbus  Category_Minivan  Category_Pickup  Category_Sedan  \\\n",
+       "15146                0.0               0.0              0.0             1.0   \n",
+       "14145                0.0               0.0              0.0             1.0   \n",
+       "8943                 0.0               0.0              0.0             1.0   \n",
+       "17889                0.0               0.0              0.0             0.0   \n",
+       "9515                 0.0               0.0              0.0             0.0   \n",
+       "\n",
+       "       ...  Manufacturer_VAZ  Manufacturer_VOLKSWAGEN  Manufacturer_VOLVO  \\\n",
+       "15146  ...               0.0                      0.0                 0.0   \n",
+       "14145  ...               0.0                      0.0                 0.0   \n",
+       "8943   ...               0.0                      0.0                 0.0   \n",
+       "17889  ...               0.0                      0.0                 0.0   \n",
+       "9515   ...               0.0                      0.0                 0.0   \n",
+       "\n",
+       "       Manufacturer_სხვა      Levy  Prod. year  Engine volume   Airbags  \\\n",
+       "15146                0.0 -1.187596    1.943625      -0.487669 -0.673150   \n",
+       "14145                0.0 -1.187596   -0.874572      -0.900100 -1.179883   \n",
+       "8943                 0.0  0.081225    0.642919      -0.487669  0.847049   \n",
+       "17889                0.0 -1.187596   -1.524925       0.474671 -0.419784   \n",
+       "9515                 0.0  1.695495    1.510056       0.474671 -0.673150   \n",
+       "\n",
+       "          Price  PricePred  \n",
+       "15146  0.144553   0.557766  \n",
+       "14145 -0.665312  -0.603334  \n",
+       "8943  -0.144678  -0.355422  \n",
+       "17889 -0.462869  -0.746737  \n",
+       "9515   2.765339   2.375842  \n",
+       "\n",
+       "[5 rows x 84 columns]"
+      ]
+     },
+     "execution_count": 59,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "pd.concat(\n",
+    "    [\n",
+    "        train_df,\n",
+    "        price_y_train,\n",
+    "        pd.Series(\n",
+    "            models[best_model][\"train_preds\"],\n",
+    "            index=price_y_train.index,\n",
+    "            name=\"PricePred\",\n",
+    "        ),\n",
+    "    ],\n",
+    "    axis=1,\n",
+    ").head(5)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 60,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>Leather interior_Yes</th>\n",
+       "      <th>Category_Coupe</th>\n",
+       "      <th>Category_Goods wagon</th>\n",
+       "      <th>Category_Hatchback</th>\n",
+       "      <th>Category_Jeep</th>\n",
+       "      <th>Category_Limousine</th>\n",
+       "      <th>Category_Microbus</th>\n",
+       "      <th>Category_Minivan</th>\n",
+       "      <th>Category_Pickup</th>\n",
+       "      <th>Category_Sedan</th>\n",
+       "      <th>...</th>\n",
+       "      <th>Manufacturer_VAZ</th>\n",
+       "      <th>Manufacturer_VOLKSWAGEN</th>\n",
+       "      <th>Manufacturer_VOLVO</th>\n",
+       "      <th>Manufacturer_სხვა</th>\n",
+       "      <th>Levy</th>\n",
+       "      <th>Prod. year</th>\n",
+       "      <th>Engine volume</th>\n",
+       "      <th>Airbags</th>\n",
+       "      <th>Price</th>\n",
+       "      <th>PricePred</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>10968</th>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>-0.055651</td>\n",
+       "      <td>-0.007434</td>\n",
+       "      <td>-1.037578</td>\n",
+       "      <td>1.353782</td>\n",
+       "      <td>-1.316082</td>\n",
+       "      <td>-0.960788</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1121</th>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>-0.331577</td>\n",
+       "      <td>0.642919</td>\n",
+       "      <td>0.474671</td>\n",
+       "      <td>1.353782</td>\n",
+       "      <td>-1.489657</td>\n",
+       "      <td>-0.235577</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>4355</th>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.398430</td>\n",
+       "      <td>1.076487</td>\n",
+       "      <td>-0.762623</td>\n",
+       "      <td>-0.673150</td>\n",
+       "      <td>0.771343</td>\n",
+       "      <td>0.887499</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>17702</th>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.528788</td>\n",
+       "      <td>-0.441003</td>\n",
+       "      <td>-0.762623</td>\n",
+       "      <td>-0.673150</td>\n",
+       "      <td>-0.679792</td>\n",
+       "      <td>-0.579671</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>6167</th>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>0.719981</td>\n",
+       "      <td>-0.441003</td>\n",
+       "      <td>-0.487669</td>\n",
+       "      <td>0.340316</td>\n",
+       "      <td>-0.173638</td>\n",
+       "      <td>-0.471104</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "<p>5 rows × 84 columns</p>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "       Leather interior_Yes  Category_Coupe  Category_Goods wagon  \\\n",
+       "10968                   1.0             0.0                   0.0   \n",
+       "1121                    1.0             0.0                   0.0   \n",
+       "4355                    1.0             0.0                   0.0   \n",
+       "17702                   1.0             0.0                   0.0   \n",
+       "6167                    0.0             0.0                   0.0   \n",
+       "\n",
+       "       Category_Hatchback  Category_Jeep  Category_Limousine  \\\n",
+       "10968                 1.0            0.0                 0.0   \n",
+       "1121                  0.0            0.0                 0.0   \n",
+       "4355                  0.0            0.0                 0.0   \n",
+       "17702                 1.0            0.0                 0.0   \n",
+       "6167                  0.0            0.0                 0.0   \n",
+       "\n",
+       "       Category_Microbus  Category_Minivan  Category_Pickup  Category_Sedan  \\\n",
+       "10968                0.0               0.0              0.0             0.0   \n",
+       "1121                 0.0               0.0              0.0             1.0   \n",
+       "4355                 0.0               0.0              0.0             1.0   \n",
+       "17702                0.0               0.0              0.0             0.0   \n",
+       "6167                 0.0               0.0              0.0             1.0   \n",
+       "\n",
+       "       ...  Manufacturer_VAZ  Manufacturer_VOLKSWAGEN  Manufacturer_VOLVO  \\\n",
+       "10968  ...               0.0                      0.0                 0.0   \n",
+       "1121   ...               0.0                      0.0                 0.0   \n",
+       "4355   ...               0.0                      0.0                 0.0   \n",
+       "17702  ...               0.0                      0.0                 0.0   \n",
+       "6167   ...               0.0                      0.0                 0.0   \n",
+       "\n",
+       "       Manufacturer_სხვა      Levy  Prod. year  Engine volume   Airbags  \\\n",
+       "10968                0.0 -0.055651   -0.007434      -1.037578  1.353782   \n",
+       "1121                 0.0 -0.331577    0.642919       0.474671  1.353782   \n",
+       "4355                 0.0  0.398430    1.076487      -0.762623 -0.673150   \n",
+       "17702                0.0  0.528788   -0.441003      -0.762623 -0.673150   \n",
+       "6167                 0.0  0.719981   -0.441003      -0.487669  0.340316   \n",
+       "\n",
+       "          Price  PricePred  \n",
+       "10968 -1.316082  -0.960788  \n",
+       "1121  -1.489657  -0.235577  \n",
+       "4355   0.771343   0.887499  \n",
+       "17702 -0.679792  -0.579671  \n",
+       "6167  -0.173638  -0.471104  \n",
+       "\n",
+       "[5 rows x 84 columns]"
+      ]
+     },
+     "execution_count": 60,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "pd.concat(\n",
+    "    [\n",
+    "        test_df,\n",
+    "        price_y_test,\n",
+    "        pd.Series(\n",
+    "            models[best_model][\"preds\"],\n",
+    "            index=price_y_test.index,\n",
+    "            name=\"PricePred\",\n",
+    "        ),\n",
+    "    ],\n",
+    "    axis=1,\n",
+    ").head(5)"
+   ]
   }
  ],
  "metadata": {