shap.datasets.rank

shap.datasets.rank()

从 LightGBM 仓库返回排名数据集。

返回:

- x_trainscipy.sparse.csr_matrix

训练特征矩阵。

- y_trainnumpy.ndarray

训练标签。

- x_testscipy.sparse.csr_matrix

测试功能矩阵。

- y_testnumpy.ndarray

测试标签。

- q_trainnumpy.ndarray

训练查询信息。

- q_testnumpy.ndarray

测试查询信息。

注释

数据源：LightGBM 仓库 https://github.com/microsoft/LightGBM/tree/master/examples/lambdarank

示例

# Usage example
x_train, y_train, x_test, y_test, q_train, q_test = shap.datasets.rank()