论文标题
生动:用于可变重要性的R包和机器学习模型的可变交互显示
vivid: An R package for Variable Importance and Variable Interactions Displays for Machine Learning Models
论文作者
论文摘要
我们提出了生动的R软件包,可在机器学习模型中观察可变重要性和可变交互。该软件包提供了一系列显示器,包括热图和基于图的显示,以查看可变重要性和互动的共同和部分依赖图,以及矩阵布局和强调重要变量子集的替代布局。为了增加机器学习模型的解释性并使该作品适用于更广泛的读者群,我们通过专注于包装结构并深入了解包装功能和关键功能,讨论了实施背后的设计选择。我们还提供了在数据集中使用的软件的实用说明。
We present vivid, an R package for visualizing variable importance and variable interactions in machine learning models. The package provides a range of displays including heatmap and graph-based displays for viewing variable importance and interaction jointly and partial dependence plots in both a matrix layout and an alternative layout emphasizing important variable subsets. With the intention of increasing a machine learning models' interpretability and making the work applicable to a wider readership, we discuss the design choices behind our implementation by focusing on the package structure and providing an in-depth look at the package functions and key features. We also provide a practical illustration of the software in use on a data set.