PeterBlake01 commited on
Commit
c915f38
1 Parent(s): 282b4d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -0
app.py CHANGED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import numpy as np
3
+ from sklearn.datasets import fetch_california_housing
4
+ from sklearn.model_selection import train_test_split
5
+ from sklearn.ensemble import RandomForestRegressor
6
+ import shap
7
+ import matplotlib.pyplot as plt
8
+
9
+ # Chargement et préparation des données California Housing
10
+ california = fetch_california_housing()
11
+ X = pd.DataFrame(california.data, columns=california.feature_names)
12
+ y = california.target # renommage pour clarté
13
+
14
+ # Division des données en ensembles d'entraînement et de test
15
+ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
16
+
17
+ # Construction et entraînement du modèle
18
+ model = RandomForestRegressor(random_state=42)
19
+ model.fit(X_train, y_train)
20
+
21
+ # Création de l'expliqueur SHAP et calcul des valeurs SHAP
22
+ explainer = shap.TreeExplainer(model)
23
+ shap_values = explainer.shap_values(X_test)
24
+
25
+ # Visualisation des valeurs SHAP
26
+ shap.summary_plot(shap_values, X_test)