Python API¶
Models¶
|
Probabilistic Labels Trees (PLTs) model with linear node estimators, using CPP core. |
|
Hierarchical Softmax model with linear node estimators, using CPP core. |
|
Binary Relevance model with linear node estimators, using CPP core |
|
One Versus Rest model with linear node estimators, using CPP core. |
Datasets¶
Load data in the libsvm format into sparse CSR matrix. |
|
|
Downloads the dataset from the internet and puts it in root directory. |
|
Downloads the dataset from the internet and puts it in root directory. |
Measures¶
|
Calculate precision at 1-k places. |
|
Calculate recall at 1-k places. |
|
Calculate coverage at 1-k places. |
|
Calculate DCG at 1-k places. |
|
Calculate nDCG at 1-k places. |
|
Calculate unnormalized hamming loss - average number of misclassified labels. |
|
Calculate inverse propensity as proposed in Jain et al. |
|
Calculate Propensity Scored Precision (PSP) at 1-k places. |