.neg_mse0s($9&9:::T)rrrdisable_default_eval_metric validation_0r%)columnszEither `group` or `qid`.)match) scipy.sparsesklearn.metricsr sklearn.model_selectionr r tmmake_ltr DataFramexgb XGBRankerfitscorecopyilocnpiscloserlenrr# evals_resultsparse csr_matrix from_spmatrixrangeshapestrpytestraises ValueError)rrscipyr r Xyq_dfrankersvalid_dfs1kfoldresultsr%r6X_csris2r s @r$run_ranking_qid_dfrU s2MsqTUVJAq!Q  BBuI ]]v; WF JJr1 RA s7NwwyHHMM$ JJr11 J/]]v; WF JJq!J b! B ::a j $U3!&"aE"&&I7|q  ;s;c;e;]]$( F  JJr11 J/ LL1 %E ::eV002>B9MbQ RRj  ''* ^^ " " 0 0 E!''!*,=>qCF>1 5 K   2q \\"a zz"a  z)C D- 2qQF8 ,--?--s/K =K##K,devicecddlm}tjdddd\}}tj j d}|jd|jd }t j|}||d <tjd | }|j|||j||}|d kDsJtjd | }||||}|D] } | dkDr Jy)z#Test LTR with categorical features.r)r i rF)rr n_categoriesonehoti)sizerT)enable_categoricalrVg?rN)r/r r0make_categoricalr9random default_rngchoicerAsortr3r4r5r6) rVr rGrHrngrltrr6scoresrMs r$run_ranking_categoricalrePs7  "1U DAq ))   %C **QQWWQZ* (C ''#,CAeH --4 ?CGGAqM IIaOE 3; --4 ?CS!Q 'F 3wr&cztjdddd\}}}}tjdd|}|j |||||fg|g|j }tjdd|d }|j |||||fg|g|j }|d d d |d d d kDsJy )zTest normalization.i@rz rank:pairwise) objectiverrV)rreval_qidF)rirrVlambdarank_normalizationr(zndcg@32r)N)r0r1r3r4r5r<)rVrGrHrrJrce0e1s r$run_normalizationrnis;;tQA.LAq#q --/& QCGGAqcaVHuG=   B --!!&  C GGAqcaVHuG=   B n i ( ,r./A)/LR/P PP Pr&)__doc__typesrtypingrnumpyr9rCxgboostr3rr0rBrUrernr&r$rusb! !@-Z@-c@-d@-FCD2QcQdQr&