Reformat NN notebook

tinkoff-ai · Jun 23, 2023 · f063c6d · f063c6d
1 parent 4268d19
commit f063c6d
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/examples/NN_examples.ipynb b/examples/NN_examples.ipynb
@@ -2566,7 +2566,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "HORIZON=7\n",
+    "HORIZON = 7\n",
     "metrics = [SMAPE(), MAPE(), MAE()]"
    ]
   },
@@ -2578,16 +2578,18 @@
    "outputs": [],
    "source": [
     "transforms = [\n",
-    "    SegmentEncoderTransform(), \n",
-    "    StandardScalerTransform(in_column='target'),\n",
-    "    DateFlagsTransform(day_number_in_week=True,\n",
-    "                       day_number_in_month=True,\n",
-    "                       week_number_in_month=False,\n",
-    "                       week_number_in_year=False,\n",
-    "                       month_number_in_year=False,\n",
-    "                       year_number=False,\n",
-    "                       is_weekend=False,\n",
-    "                       out_column='df'),       \n",
+    "    SegmentEncoderTransform(),\n",
+    "    StandardScalerTransform(in_column=\"target\"),\n",
+    "    DateFlagsTransform(\n",
+    "        day_number_in_week=True,\n",
+    "        day_number_in_month=True,\n",
+    "        week_number_in_month=False,\n",
+    "        week_number_in_year=False,\n",
+    "        month_number_in_year=False,\n",
+    "        year_number=False,\n",
+    "        is_weekend=False,\n",
+    "        out_column=\"df\",\n",
+    "    ),\n",
     "]"
    ]
   },
@@ -2598,7 +2600,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "monthly_smm = SeasonalitySSM(num_seasons=31, timestamp_transform=lambda x: x.day-1)\n",
+    "monthly_smm = SeasonalitySSM(num_seasons=31, timestamp_transform=lambda x: x.day - 1)\n",
     "weekly_smm = SeasonalitySSM(num_seasons=7, timestamp_transform=lambda x: x.weekday())"
    ]
   },