Text

Visualizing Variable Importance and Variable Interaction Effects in Machine Learning Models

Variable importance, interaction measures, and partial dependence plots are important summaries in the interpretation of statistical and machine learning models. In this article, we describe new visualization techniques for exploring these model summaries. We construct heatmap and graph-based displays showing variable importance and interaction jointly, which are carefully designed to highlight important aspects of the fit. We describe a new matrix-type layout showing all single and bivariate partial dependence plots, and an alternative layout based on graph Eulerians focusing on key subsets. Our new visualizations are model-agnostic and are applicable to regression and classification supervised learning settings. They enhance interpretation even in situations where the number of variables is large. Our R package vivid (variable importance and variable interaction displays) provides an implementation. Supplementary files for this article are available online.

Ketersediaan

Informasi Detail

Judul Seri

JOURNAL OF COMPUTATIONAL AND GRAPHICAL STATISTICS; Vol.31 No.3 September 2022

No. Panggil

-

Penerbit

: .,

Deskripsi Fisik

p. 766-778

Bahasa

English

ISBN/ISSN

-

Klasifikasi

NONE

Tipe Isi

-

Tipe Media

-

Tipe Pembawa

-

Edisi

-

Subjek

MODEL VISUALIZATION
MODEL EXPLANATION
BLACK-BOX

Info Detail Spesifik

https://doi.org/10.1080/10618600.2021.2007935

Pernyataan Tanggungjawab

Alan Inglis, Andrew Parnell, Catherine B. Hurley

Versi lain/terkait

Lampiran Berkas

Komentar