Update ml-regression.md

Author: LiamConnors

doc/python/ml-regression.md

@@ -6,9 +6,9 @@ jupyter:
       extension: .md
       format_name: markdown
       format_version: '1.3'
-      jupytext_version: 1.14.1
+      jupytext_version: 1.16.1
   kernelspec:
-    display_name: Python 3
+    display_name: Python 3 (ipykernel)
     language: python
     name: python3
   language_info:
@@ -20,7 +20,7 @@ jupyter:
     name: python
     nbconvert_exporter: python
     pygments_lexer: ipython3
-    version: 3.8.8
+    version: 3.10.0
   plotly:
     description: Visualize regression in scikit-learn with Plotly.
     display_as: ai_ml
@@ -412,6 +412,7 @@ import pandas as pd
 import plotly.express as px
 import plotly.graph_objects as go
 from sklearn.linear_model import LassoCV
+from sklearn.preprocessing import StandardScaler
 
 N_FOLD = 6
 
@@ -421,9 +422,13 @@ X = df.drop(columns=['lifeExp', 'iso_num'])
 X = pd.get_dummies(X, columns=['country', 'continent', 'iso_alpha'])
 y = df['lifeExp']
 
+# Normalize the data
+scaler = StandardScaler()
+X_scaled = scaler.fit_transform(X)
+
 # Train model to predict life expectancy
-model = LassoCV(cv=N_FOLD, normalize=True)
-model.fit(X, y)
+model = LassoCV(cv=N_FOLD)
+model.fit(X_scaled, y)
 mean_alphas = model.mse_path_.mean(axis=-1)
 
 fig = go.Figure([