linkedin / linkedin/FastTreeSHAP

FastTreeSHAP summary_plot plots interaction value instead of impact on model output

Aperta
#29 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Python
Stelle
566
Fork
38
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

I am using FastTreeSHAP to help explained my random forest model prediction having over 40K samples and 24 features.

explainer = fasttreeshap.TreeExplainer(model, algorithm='auto' ,n_jobs=-1)
shap_values = explainer(X_test).values

shap_values.shape
(40682, 24, 5)

# plotting
fasttreeshap.summary_plot(shap_values, X_test, plot_type = 'bar')

The output figure is (showing top-5 features interaction):

Screenshot 2023-07-05 at 14 50 08

I expected output figure to look similar to below image (FastTreeSHAP docs).

Screenshot 2023-07-05 at 14 51 23

Is there any change in the summary_plot() method?

Guida per i contributori

Apri la guida per i contributori

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia con fasttreeshap.summary_plot() utilizzando la forma 3-D riprodotta di shap_values e l'esempio di grafico a barre nell'issue. Confronta la classificazione delle feature generata con l'output atteso documentato, quindi verifica che il comportamento completato rappresenti l'impatto sull'output del modello anziché i valori di interazione.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
python
Ambito
data-visualization, machine-learning
Tipo di issue
Bug
Difficoltà
3/5
Tempo stimato
1-2 giorni
Stato di attività
Ferma
Chiarezza
Abbastanza chiara
Idoneità per principianti
48/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.