From b04d33095166f8d5f323fc7cdb85404bff1faa49 Mon Sep 17 00:00:00 2001
From: Alexander Grote <grote@fzi.de>
Date: Thu, 25 Apr 2024 15:11:18 +0200
Subject: [PATCH] Updating regression exercise

---
 ml_bootcamp/Regression_Exercise.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ml_bootcamp/Regression_Exercise.ipynb b/ml_bootcamp/Regression_Exercise.ipynb
index 73e6324..1e26d26 100644
--- a/ml_bootcamp/Regression_Exercise.ipynb
+++ b/ml_bootcamp/Regression_Exercise.ipynb
@@ -280,7 +280,7 @@
     "missing_data.head(20)\n",
     "\n",
     "f, ax = plt.subplots()\n",
-    "plt.xticks(rotation='90')\n",
+    "plt.xticks(rotation='vertical')\n",
     "sns.barplot(x=X_train_na.index, y=X_train_na)\n",
     "plt.xlabel('Features', fontsize=15)\n",
     "plt.ylabel('Percent of missing values', fontsize=15)\n",
@@ -295,7 +295,7 @@
    "outputs": [],
    "source": [
     "## visualize correlation\n",
-    "sns.heatmap(X_train.corr())\n",
+    "sns.heatmap(X_train.select_dtypes(np.number).corr())\n",
     "plt.show()"
    ]
   },
-- 
GitLab