shap.datasets.rank

shap.datasets.rank() → tuple[ssp.csr_matrix, np.ndarray, ssp.csr_matrix, np.ndarray, np.ndarray, np.ndarray]

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

用于排序任务。

返回:

x_trainscipy.sparse.csr_matrix

训练特征矩阵。

y_trainnumpy.ndarray

训练标签。

x_testscipy.sparse.csr_matrix

测试特征矩阵。

y_testnumpy.ndarray

测试标签。

q_trainnumpy.ndarray

训练查询信息。

q_testnumpy.ndarray

测试查询信息。

注释

数据来源: LightGBM repository https://github.com/microsoft/LightGBM/tree/master/examples/lambdarank

示例

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