From 7f542d2198a2e00310b6d576855dd931ab8b4da6 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Fri, 31 Jan 2020 18:46:13 +0800 Subject: [PATCH] Test model compatibility. (#5248) * Add model compability tests. * Typo. --- src/common/json.cc | 4 +- tests/python/generate_models.py | 124 ++++++++++++++++++ tests/python/models/version | 1 + .../models/xgboost-1.0.0-SNAPSHOT.cls.bin | Bin 0 -> 4891 bytes .../models/xgboost-1.0.0-SNAPSHOT.cls.json | 1 + .../models/xgboost-1.0.0-SNAPSHOT.ltr.bin | Bin 0 -> 1799 bytes .../models/xgboost-1.0.0-SNAPSHOT.ltr.json | 1 + .../models/xgboost-1.0.0-SNAPSHOT.reg.bin | Bin 0 -> 1950 bytes .../models/xgboost-1.0.0-SNAPSHOT.reg.json | 1 + .../xgboost_scikit-1.0.0-SNAPSHOT.cls.bin | Bin 0 -> 5641 bytes .../xgboost_scikit-1.0.0-SNAPSHOT.cls.json | 1 + .../xgboost_scikit-1.0.0-SNAPSHOT.ltr.bin | Bin 0 -> 2445 bytes .../xgboost_scikit-1.0.0-SNAPSHOT.ltr.json | 1 + .../xgboost_scikit-1.0.0-SNAPSHOT.reg.bin | Bin 0 -> 2606 bytes .../xgboost_scikit-1.0.0-SNAPSHOT.reg.json | 1 + tests/python/test_model_compatibility.py | 47 +++++++ tests/python/test_ranking.py | 6 +- 17 files changed, 183 insertions(+), 5 deletions(-) create mode 100644 tests/python/generate_models.py create mode 100644 tests/python/models/version create mode 100644 tests/python/models/xgboost-1.0.0-SNAPSHOT.cls.bin create mode 100644 tests/python/models/xgboost-1.0.0-SNAPSHOT.cls.json create mode 100644 tests/python/models/xgboost-1.0.0-SNAPSHOT.ltr.bin create mode 100644 tests/python/models/xgboost-1.0.0-SNAPSHOT.ltr.json create mode 100644 tests/python/models/xgboost-1.0.0-SNAPSHOT.reg.bin create mode 100644 tests/python/models/xgboost-1.0.0-SNAPSHOT.reg.json create mode 100644 tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.cls.bin create mode 100644 tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.cls.json create mode 100644 tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.ltr.bin create mode 100644 tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.ltr.json create mode 100644 tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.reg.bin create mode 100644 tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.reg.json create mode 100644 tests/python/test_model_compatibility.py diff --git a/src/common/json.cc b/src/common/json.cc index ecdcce3d3..52878bbf9 100644 --- a/src/common/json.cc +++ b/src/common/json.cc @@ -24,7 +24,7 @@ void JsonWriter::Visit(JsonArray const* arr) { for (size_t i = 0; i < size; ++i) { auto const& value = vec[i]; this->Save(value); - if (i != size-1) { Write(", "); } + if (i != size-1) { Write(","); } } this->Write("]"); } @@ -38,7 +38,7 @@ void JsonWriter::Visit(JsonObject const* obj) { size_t size = obj->getObject().size(); for (auto& value : obj->getObject()) { - this->Write("\"" + value.first + "\": "); + this->Write("\"" + value.first + "\":"); this->Save(value.second); if (i != size-1) { diff --git a/tests/python/generate_models.py b/tests/python/generate_models.py new file mode 100644 index 000000000..4f02566a5 --- /dev/null +++ b/tests/python/generate_models.py @@ -0,0 +1,124 @@ +import xgboost +import numpy as np +import os + +kRounds = 2 +kRows = 1000 +kCols = 4 +kForests = 2 +kMaxDepth = 2 +kClasses = 3 + +X = np.random.randn(kRows, kCols) +w = np.random.uniform(size=kRows) + +version = xgboost.__version__ + +np.random.seed(1994) +target_dir = 'models' + + +def booster_bin(model): + return os.path.join(target_dir, + 'xgboost-' + version + '.' + model + '.bin') + + +def booster_json(model): + return os.path.join(target_dir, + 'xgboost-' + version + '.' + model + '.json') + + +def skl_bin(model): + return os.path.join(target_dir, + 'xgboost_scikit-' + version + '.' + model + '.bin') + + +def skl_json(model): + return os.path.join(target_dir, + 'xgboost_scikit-' + version + '.' + model + '.json') + + +def generate_regression_model(): + print('Regression') + y = np.random.randn(kRows) + + data = xgboost.DMatrix(X, label=y, weight=w) + booster = xgboost.train({'tree_method': 'hist', + 'num_parallel_tree': kForests, + 'max_depth': kMaxDepth}, + num_boost_round=kRounds, dtrain=data) + booster.save_model(booster_bin('reg')) + booster.save_model(booster_json('reg')) + + reg = xgboost.XGBRegressor(tree_method='hist', + num_parallel_tree=kForests, + max_depth=kMaxDepth, + n_estimators=kRounds) + reg.fit(X, y, w) + reg.save_model(skl_bin('reg')) + reg.save_model(skl_json('reg')) + + +def generate_classification_model(): + print('Classification') + y = np.random.randint(0, kClasses, size=kRows) + data = xgboost.DMatrix(X, label=y, weight=w) + booster = xgboost.train({'num_class': kClasses, + 'tree_method': 'hist', + 'num_parallel_tree': kForests, + 'max_depth': kMaxDepth}, + num_boost_round=kRounds, dtrain=data) + booster.save_model(booster_bin('cls')) + booster.save_model(booster_json('cls')) + + cls = xgboost.XGBClassifier(tree_method='hist', + num_parallel_tree=kForests, + max_depth=kMaxDepth, + n_estimators=kRounds) + cls.fit(X, y, w) + cls.save_model(skl_bin('cls')) + cls.save_model(skl_json('cls')) + + +def generate_ranking_model(): + print('Learning to Rank') + y = np.random.randint(5, size=kRows) + w = np.random.randn(20) + g = np.repeat(50, 20) + + data = xgboost.DMatrix(X, y, weight=w) + data.set_group(g) + booster = xgboost.train({'objective': 'rank:ndcg', + 'num_parallel_tree': kForests, + 'tree_method': 'hist', + 'max_depth': kMaxDepth}, + num_boost_round=kRounds, + dtrain=data) + booster.save_model(booster_bin('ltr')) + booster.save_model(booster_json('ltr')) + + ranker = xgboost.sklearn.XGBRanker(n_estimators=kRounds, + tree_method='hist', + objective='rank:ndcg', + max_depth=kMaxDepth, + num_parallel_tree=kForests) + ranker.fit(X, y, g, sample_weight=w) + ranker.save_model(skl_bin('ltr')) + ranker.save_model(skl_json('ltr')) + + +def write_versions(): + versions = {'numpy': np.__version__, + 'xgboost': version} + with open(os.path.join(target_dir, 'version'), 'w') as fd: + fd.write(str(versions)) + + +if __name__ == '__main__': + if not os.path.exists(target_dir): + os.mkdir(target_dir) + + generate_regression_model() + generate_classification_model() + generate_ranking_model() + write_versions() diff --git a/tests/python/models/version b/tests/python/models/version new file mode 100644 index 000000000..194fa40f4 --- /dev/null +++ b/tests/python/models/version @@ -0,0 +1 @@ +{'numpy': '1.16.4', 'xgboost': '1.0.0-SNAPSHOT'} \ No newline at end of file diff --git a/tests/python/models/xgboost-1.0.0-SNAPSHOT.cls.bin b/tests/python/models/xgboost-1.0.0-SNAPSHOT.cls.bin new file mode 100644 index 0000000000000000000000000000000000000000..81029ac7a6e80fbb2798c0ad4f5d506dd8704859 GIT binary patch literal 4891 zcmZQzV6bOlU|?Vd;(-LbP|I^mb4oI;iu2P-auX}qpn~a1B}J*JJU|}fK)R8D-5|HI zBdGxe>p+A5Kmc+V6A**w-xI8DfK&r0x?%21Q%<)7av4~G7#8Ao=lKo*)i;3jg8)c= zeazoIATuB$5V3=aV%9*jQPdT)%-svJ4_V!_Df4Y=pISNS1SU8yuJPUtGEXgcgZ)m1 zr%t<#UxC$^-6TaNT6u58?>KZ?qSWy6zAAQ1yDdl#mS;ffIajIgM^ECBq$HZ!CpQr_%R<;mch$wP~0cxa)Qe{Sa}T+E0s642hu3%HczN!H%Jb- z9G08F3@Teu)Ghd>g;8D?rEa!rzb6SUudSai2A8*ABA6Yv1@buiG@k*N&z$dX+rRy{ z-05HXbFhBj`5q4QPPIC1m%FnEXe~pToTbCr87-7b>cb*m3W8;P844D>(nwFW79~vM$}x{wO=x zy%~>I+MAt`cNV=h7o30oAKPHh=%wwnY^x_Y{|trvG1B5=DAvPezpkUSXFzGyUhs)M zdb;Fb=#0_&kV+J9PIUHL=wXRU~+?(*Pt@(5~CDI z4Br0)rAIF>9gOli{%rw9|C8zMLyY!}+v|OxJ}F9hz1wiXp3FdLhhJ9{oDY^m`j2x2 zw%d1}<#uFw#{f$E42Kow+AB#$I2$P32B-UN(TDBko>6r;fAj=6-3GkaW_NhY5r;-s z8*n@(l~f3ci%s-!$PKS|%2X5Td>&FNSjE4Vx#3gz`kiVt8Eg8?XBKo}I~APfWO F;sB-Y&Ex<8 literal 0 HcmV?d00001 diff --git a/tests/python/models/xgboost-1.0.0-SNAPSHOT.cls.json b/tests/python/models/xgboost-1.0.0-SNAPSHOT.cls.json new file mode 100644 index 000000000..e8bc8bceb --- /dev/null +++ b/tests/python/models/xgboost-1.0.0-SNAPSHOT.cls.json @@ -0,0 +1 @@ +{"learner":{"attributes":{},"gradient_booster":{"model":{"gbtree_model_param":{"num_trees":"12","size_leaf_vector":"0"},"tree_info":[0,0,1,1,2,2,0,0,1,1,2,2],"trees":[{"base_weights":[-3.10496310703456402e-03,-1.14356480538845062e-01,1.12878747284412384e-01,-2.06186547875404358e-01,1.52976317331194878e-02,-5.11124253273010254e-01,1.29867866635322571e-01],"default_left":[false,false,false,false,false,false,false],"id":0,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.92025685310363770e+00,1.37833094596862793e+00,1.24116408824920654e+00,1.33592224121093750e+00,1.26208007335662842e+00,0.00000000000000000e+00,1.38571357727050781e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[1.14481402561068535e-02,2.25720971822738647e-01,-2.23139905929565430e+00,-3.09279840439558029e-02,2.29464494623243809e-03,-7.66686424612998962e-02,1.94801799952983856e-02],"split_indices":[1,0,0,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.14501411437988281e+02,1.09815666198730469e+02,6.66821746826171875e+01,4.78192405700683594e+01,2.13970732688903809e+00,1.07675956726074219e+02],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.10496310703456402e-03,-1.14356480538845062e-01,1.12878747284412384e-01,-2.06186547875404358e-01,1.52976317331194878e-02,-5.11124253273010254e-01,1.29867866635322571e-01],"default_left":[false,false,false,false,false,false,false],"id":1,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.92025685310363770e+00,1.37833094596862793e+00,1.24116408824920654e+00,1.33592224121093750e+00,1.26208007335662842e+00,0.00000000000000000e+00,1.38571357727050781e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[1.14481402561068535e-02,2.25720971822738647e-01,-2.23139905929565430e+00,-3.09279840439558029e-02,2.29464494623243809e-03,-7.66686424612998962e-02,1.94801799952983856e-02],"split_indices":[1,0,0,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.14501411437988281e+02,1.09815666198730469e+02,6.66821746826171875e+01,4.78192405700683594e+01,2.13970732688903809e+00,1.07675956726074219e+02],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-1.69028975069522858e-02,8.05607438087463379e-01,-2.48735304921865463e-02,1.18516474962234497e-01,-7.35604763031005859e-02],"default_left":[false,false,false,false,false],"id":2,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.49720096588134766e+00,0.00000000000000000e+00,1.57338166236877441e+00,1.47135305404663086e+00,1.12515950202941895e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.20841123163700104e-01,-6.54938280582427979e-01,1.77774727344512939e-02,-1.10340714454650879e-02],"split_indices":[0,0,1,0,0],"sum_hessian":[2.24317077636718750e+02,1.19245672225952148e+00,2.23124618530273438e+02,5.61932411193847656e+01,1.66931381225585938e+02],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[-1.69028975069522858e-02,8.05607438087463379e-01,-2.48735304921865463e-02,1.18516474962234497e-01,-7.35604763031005859e-02],"default_left":[false,false,false,false,false],"id":3,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.49720096588134766e+00,0.00000000000000000e+00,1.57338166236877441e+00,1.47135305404663086e+00,1.12515950202941895e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.20841123163700104e-01,-6.54938280582427979e-01,1.77774727344512939e-02,-1.10340714454650879e-02],"split_indices":[0,0,1,0,0],"sum_hessian":[2.24317077636718750e+02,1.19245672225952148e+00,2.23124618530273438e+02,5.61932411193847656e+01,1.66931381225585938e+02],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[2.00077202171087265e-02,9.98598039150238037e-02,-1.12991318106651306e-01,-2.68701147288084030e-02,2.14132413268089294e-01,-2.78286129236221313e-01,9.12456586956977844e-03],"default_left":[false,false,false,false,false,false,false],"id":4,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.40420699119567871e+00,2.05230426788330078e+00,1.71823227405548096e+00,3.13472294807434082e+00,2.89248418807983398e+00,9.98327255249023438e-01,1.37142074108123779e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[3.12443733215332031e-01,-5.36825239658355713e-01,6.99698746204376221e-01,-4.03051730245351791e-03,3.21198627352714539e-02,-4.17429208755493164e-02,1.36868492700159550e-03],"split_indices":[1,1,1,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.40319183349609375e+02,8.39979019165039062e+01,6.68957366943359375e+01,7.34234390258789062e+01,3.51459007263183594e+01,4.88520011901855469e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[2.00077202171087265e-02,9.98598039150238037e-02,-1.12991318106651306e-01,-2.68701147288084030e-02,2.14132413268089294e-01,-2.78286129236221313e-01,9.12456586956977844e-03],"default_left":[false,false,false,false,false,false,false],"id":5,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.40420699119567871e+00,2.05230426788330078e+00,1.71823227405548096e+00,3.13472294807434082e+00,2.89248418807983398e+00,9.98327255249023438e-01,1.37142074108123779e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[3.12443733215332031e-01,-5.36825239658355713e-01,6.99698746204376221e-01,-4.03051730245351791e-03,3.21198627352714539e-02,-4.17429208755493164e-02,1.36868492700159550e-03],"split_indices":[1,1,1,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.40319183349609375e+02,8.39979019165039062e+01,6.68957366943359375e+01,7.34234390258789062e+01,3.51459007263183594e+01,4.88520011901855469e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.14401555806398392e-03,-2.77371466159820557e-01,2.71770134568214417e-02,-5.22169768810272217e-01,-1.32588371634483337e-01,-1.92690286785364151e-02,1.96981996297836304e-01],"default_left":[false,false,false,false,false,false,false],"id":6,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.87940013408660889e+00,7.28317379951477051e-01,1.60969936847686768e+00,3.83746862411499023e-01,1.40333545207977295e+00,1.32722187042236328e+00,1.38107287883758545e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-1.36123311519622803e+00,-3.55613559484481812e-01,8.09594929218292236e-01,-7.83254653215408325e-02,-1.98882557451725006e-02,-2.89035448804497719e-03,2.95473001897335052e-02],"split_indices":[1,2,0,0,0,0,0],"sum_hessian":[2.24122543334960938e+02,2.15025730133056641e+01,2.02619964599609375e+02,7.02246618270874023e+00,1.44801063537597656e+01,1.59797698974609375e+02,4.28222694396972656e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.14401555806398392e-03,-2.77371466159820557e-01,2.71770134568214417e-02,-5.22169768810272217e-01,-1.32588371634483337e-01,-1.92690286785364151e-02,1.96981996297836304e-01],"default_left":[false,false,false,false,false,false,false],"id":7,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.87940013408660889e+00,7.28317379951477051e-01,1.60969936847686768e+00,3.83746862411499023e-01,1.40333545207977295e+00,1.32722187042236328e+00,1.38107287883758545e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-1.36123311519622803e+00,-3.55613559484481812e-01,8.09594929218292236e-01,-7.83254653215408325e-02,-1.98882557451725006e-02,-2.89035448804497719e-03,2.95473001897335052e-02],"split_indices":[1,2,0,0,0,0,0],"sum_hessian":[2.24122543334960938e+02,2.15025730133056641e+01,2.02619964599609375e+02,7.02246618270874023e+00,1.44801063537597656e+01,1.59797698974609375e+02,4.28222694396972656e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-1.22269820421934128e-02,7.03054547309875488e-01,-1.94818396121263504e-02,-3.24203968048095703e-02,3.11440438032150269e-01],"default_left":[false,false,false,false,false],"id":8,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.17855262756347656e+00,0.00000000000000000e+00,9.65734302997589111e-01,1.15091991424560547e+00,1.31937885284423828e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.05458185076713562e-01,1.78177630901336670e+00,-4.86305961385369301e-03,4.67160679399967194e-02],"split_indices":[0,0,2,0,0],"sum_hessian":[2.23474166870117188e+02,1.28086733818054199e+00,2.22193298339843750e+02,2.14700851440429688e+02,7.49244642257690430e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[-1.22269820421934128e-02,7.03054547309875488e-01,-1.94818396121263504e-02,-3.24203968048095703e-02,3.11440438032150269e-01],"default_left":[false,false,false,false,false],"id":9,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.17855262756347656e+00,0.00000000000000000e+00,9.65734302997589111e-01,1.15091991424560547e+00,1.31937885284423828e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.05458185076713562e-01,1.78177630901336670e+00,-4.86305961385369301e-03,4.67160679399967194e-02],"split_indices":[0,0,2,0,0],"sum_hessian":[2.23474166870117188e+02,1.28086733818054199e+00,2.22193298339843750e+02,2.14700851440429688e+02,7.49244642257690430e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[1.52873406186699867e-02,5.03631830215454102e-01,-6.68718665838241577e-03,-9.80335399508476257e-02,7.35435545444488525e-01,3.88425774872303009e-02,-1.30366414785385132e-01],"default_left":[false,false,false,false,false,false,false],"id":10,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.42684388160705566e+00,1.43019437789916992e+00,1.22756230831146240e+00,3.81559073925018311e-01,3.57241153717041016e-01,1.24453198909759521e+00,1.17859208583831787e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-2.00600242614746094e+00,-2.40636110305786133e+00,6.73008561134338379e-01,-1.47050311788916588e-02,1.10315337777137756e-01,5.82638662308454514e-03,-1.95549633353948593e-02],"split_indices":[1,1,2,0,0,0,0],"sum_hessian":[2.24835845947265625e+02,8.73767948150634766e+00,2.16098175048828125e+02,2.59061360359191895e+00,6.14706563949584961e+00,1.58453109741210938e+02,5.76450576782226562e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[1.52873406186699867e-02,5.03631830215454102e-01,-6.68718665838241577e-03,-9.80335399508476257e-02,7.35435545444488525e-01,3.88425774872303009e-02,-1.30366414785385132e-01],"default_left":[false,false,false,false,false,false,false],"id":11,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.42684388160705566e+00,1.43019437789916992e+00,1.22756230831146240e+00,3.81559073925018311e-01,3.57241153717041016e-01,1.24453198909759521e+00,1.17859208583831787e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-2.00600242614746094e+00,-2.40636110305786133e+00,6.73008561134338379e-01,-1.47050311788916588e-02,1.10315337777137756e-01,5.82638662308454514e-03,-1.95549633353948593e-02],"split_indices":[1,1,2,0,0,0,0],"sum_hessian":[2.24835845947265625e+02,8.73767948150634766e+00,2.16098175048828125e+02,2.59061360359191895e+00,6.14706563949584961e+00,1.58453109741210938e+02,5.76450576782226562e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"0.500000","num_class":"3","num_feature":"4"},"objective":{"name":"multi:softmax","softmax_multiclass_param":{"num_class":"3"}}},"version":[1,0,0]} \ No newline at end of file diff --git a/tests/python/models/xgboost-1.0.0-SNAPSHOT.ltr.bin b/tests/python/models/xgboost-1.0.0-SNAPSHOT.ltr.bin new file mode 100644 index 0000000000000000000000000000000000000000..c99505b854146d882bece5fee9cdc2f21cb28f05 GIT binary patch literal 1799 zcmZQzV6bOl7+DPR6)z6#1DShMZ?D}|hUJbi3#@EG`j3l9+Hd~Z>iD|- zsVzu8fl1cBOQywvThR@yel#4Z7mn{mS5@0c$uZwRt4 zXU0=IriD`-vvxn*3sQfi_m|yNzLgH|pKP%M@j;;Y9=}88>*wHf>KWf-XR@@}k!!U+ pIRAj;^HrDFckKEM=UcKiI2cR(w*$$6+&UT`3=D%MK0tAT3;_SxINksN literal 0 HcmV?d00001 diff --git a/tests/python/models/xgboost-1.0.0-SNAPSHOT.ltr.json b/tests/python/models/xgboost-1.0.0-SNAPSHOT.ltr.json new file mode 100644 index 000000000..a19374c34 --- /dev/null +++ b/tests/python/models/xgboost-1.0.0-SNAPSHOT.ltr.json @@ -0,0 +1 @@ +{"learner":{"attributes":{},"gradient_booster":{"model":{"gbtree_model_param":{"num_trees":"4","size_leaf_vector":"0"},"tree_info":[0,0,0,0],"trees":[{"base_weights":[2.18596185597164094e-09,-3.76773595809936523e-01,4.55630868673324585e-02,1.12075649201869965e-01,-1.93485423922538757e-01],"default_left":[false,false,false,false,false],"id":0,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[4.20947641134262085e-01,0.00000000000000000e+00,3.69498044252395630e-01,5.97973287105560303e-01,6.13317489624023438e-01],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-1.45796775817871094e+00,-5.65160401165485382e-02,8.68250608444213867e-01,1.68113484978675842e-02,-2.90228147059679031e-02],"split_indices":[3,0,1,0,0],"sum_hessian":[2.25207920074462891e+01,1.64538443088531494e+00,2.08754062652587891e+01,1.67469234466552734e+01,4.12848377227783203e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[2.18596185597164094e-09,-3.76773595809936523e-01,4.55630868673324585e-02,1.12075649201869965e-01,-1.93485423922538757e-01],"default_left":[false,false,false,false,false],"id":1,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[4.20947641134262085e-01,0.00000000000000000e+00,3.69498044252395630e-01,5.97973287105560303e-01,6.13317489624023438e-01],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-1.45796775817871094e+00,-5.65160401165485382e-02,8.68250608444213867e-01,1.68113484978675842e-02,-2.90228147059679031e-02],"split_indices":[3,0,1,0,0],"sum_hessian":[2.25207920074462891e+01,1.64538443088531494e+00,2.08754062652587891e+01,1.67469234466552734e+01,4.12848377227783203e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[2.31542762740843955e-09,-1.12662151455879211e-01,3.53309124708175659e-01,-4.52967911958694458e-01,-4.28877249360084534e-02,-1.19008123874664307e-01,4.98231500387191772e-01],"default_left":[false,false,false,false,false,false,false],"id":2,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.03438735008239746e+00,4.48428511619567871e-01,4.89362835884094238e-01,0.00000000000000000e+00,2.74164468050003052e-01,0.00000000000000000e+00,0.00000000000000000e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[5.69312453269958496e-01,-1.49666213989257812e+00,-3.32068562507629395e-01,-6.79451897740364075e-02,-6.43315911293029785e-03,-1.78512185811996460e-02,7.47347250580787659e-02],"split_indices":[1,1,0,0,0,0,0],"sum_hessian":[2.39866485595703125e+01,1.87036170959472656e+01,5.28303003311157227e+00,2.24795222282409668e+00,1.64556655883789062e+01,1.28239238262176514e+00,4.00063753128051758e+00],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[2.31542762740843955e-09,-1.12662151455879211e-01,3.53309124708175659e-01,-4.52967911958694458e-01,-4.28877249360084534e-02,-1.19008123874664307e-01,4.98231500387191772e-01],"default_left":[false,false,false,false,false,false,false],"id":3,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.03438735008239746e+00,4.48428511619567871e-01,4.89362835884094238e-01,0.00000000000000000e+00,2.74164468050003052e-01,0.00000000000000000e+00,0.00000000000000000e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[5.69312453269958496e-01,-1.49666213989257812e+00,-3.32068562507629395e-01,-6.79451897740364075e-02,-6.43315911293029785e-03,-1.78512185811996460e-02,7.47347250580787659e-02],"split_indices":[1,1,0,0,0,0,0],"sum_hessian":[2.39866485595703125e+01,1.87036170959472656e+01,5.28303003311157227e+00,2.24795222282409668e+00,1.64556655883789062e+01,1.28239238262176514e+00,4.00063753128051758e+00],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"0.500000","num_class":"0","num_feature":"4"},"objective":{"lambda_rank_param":{"fix_list_weight":"0","num_pairsample":"1"},"name":"rank:ndcg"}},"version":[1,0,0]} \ No newline at end of file diff --git a/tests/python/models/xgboost-1.0.0-SNAPSHOT.reg.bin b/tests/python/models/xgboost-1.0.0-SNAPSHOT.reg.bin new file mode 100644 index 0000000000000000000000000000000000000000..7baeaf373a23847ea474c5d72aed5a2f7bbd30f7 GIT binary patch literal 1950 zcmZQzV6bOl7a#uX<|`oN@`J2ei0i~JUywTC^Z%2QpSOIEKV1I+{TV1 zI~YRtKM;W2#RSA4+PuNZ4oEdH0|}V>_B?&DAIN231!7o;zuLj>094-q(hmY4`NYEx z`#@$uL?B}PR#kg}W}~Rn&bbT{Ls1ucvCo#_=6Q$U=s(UHi#fseDX3p`sCY5K*=WQ4 zeIWG?$&(zmJ`Hn@+NZrA#AmpbZI}C;%kjdZ1z`0b4xe$bo!H~Nz4gLAko1LDJq~WX zAqB=h@5HP<^0{j>(EDA u5MO3SyF)=@nlszanPBrK1Wj|;W!>U@{rLwlALQ1-oIgfde1PH%82|v;MQ#cJ literal 0 HcmV?d00001 diff --git a/tests/python/models/xgboost-1.0.0-SNAPSHOT.reg.json b/tests/python/models/xgboost-1.0.0-SNAPSHOT.reg.json new file mode 100644 index 000000000..b6890cea0 --- /dev/null +++ b/tests/python/models/xgboost-1.0.0-SNAPSHOT.reg.json @@ -0,0 +1 @@ +{"learner":{"attributes":{},"gradient_booster":{"model":{"gbtree_model_param":{"num_trees":"4","size_leaf_vector":"0"},"tree_info":[0,0,0,0],"trees":[{"base_weights":[-5.37645816802978516e-01,-4.36891138553619385e-01,-6.70873284339904785e-01,-1.25496864318847656e+00,-4.07270163297653198e-01,-6.88224375247955322e-01,4.64901357889175415e-01],"default_left":[false,false,false,false,false,false,false],"id":0,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[6.49523925781250000e+00,6.53602600097656250e+00,4.57461547851562500e+00,2.30323791503906250e-01,6.39891815185546875e+00,4.40366363525390625e+00,2.28362298011779785e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[1.89942225813865662e-01,-1.81951093673706055e+00,2.12066125869750977e+00,-1.88245311379432678e-01,-6.10905252397060394e-02,-1.03233657777309418e-01,6.97352066636085510e-02],"split_indices":[1,0,0,0,0,0,0],"sum_hessian":[5.04713470458984375e+02,2.89816162109375000e+02,2.14897293090820312e+02,8.68150043487548828e+00,2.81134674072265625e+02,2.12051849365234375e+02,2.84543561935424805e+00],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-5.37645816802978516e-01,-4.36891138553619385e-01,-6.70873284339904785e-01,-1.25496864318847656e+00,-4.07270163297653198e-01,-6.88224375247955322e-01,4.64901357889175415e-01],"default_left":[false,false,false,false,false,false,false],"id":1,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[6.49523925781250000e+00,6.53602600097656250e+00,4.57461547851562500e+00,2.30323791503906250e-01,6.39891815185546875e+00,4.40366363525390625e+00,2.28362298011779785e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[1.89942225813865662e-01,-1.81951093673706055e+00,2.12066125869750977e+00,-1.88245311379432678e-01,-6.10905252397060394e-02,-1.03233657777309418e-01,6.97352066636085510e-02],"split_indices":[1,0,0,0,0,0,0],"sum_hessian":[5.04713470458984375e+02,2.89816162109375000e+02,2.14897293090820312e+02,8.68150043487548828e+00,2.81134674072265625e+02,2.12051849365234375e+02,2.84543561935424805e+00],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.77470612525939941e-01,3.31088960170745850e-01,-3.92237067222595215e-01,8.17872881889343262e-01,1.18046358227729797e-01,-3.00728023052215576e-01,-4.70518797636032104e-01],"default_left":[false,false,false,false,false,false,false],"id":2,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[5.42109680175781250e+00,1.03034389019012451e+00,3.41049194335937500e+00,0.00000000000000000e+00,1.19803142547607422e+00,4.23731803894042969e+00,4.69757843017578125e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-2.07929229736328125e+00,-5.09094715118408203e-01,-8.72411578893661499e-02,1.22680939733982086e-01,1.77069548517465591e-02,-4.51092049479484558e-02,-7.05778226256370544e-02],"split_indices":[3,0,3,0,0,0,0],"sum_hessian":[5.04713470458984375e+02,9.86623668670654297e+00,4.94847229003906250e+02,2.13924217224121094e+00,7.72699451446533203e+00,2.30380615234375000e+02,2.64466613769531250e+02],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.77470612525939941e-01,3.31088960170745850e-01,-3.92237067222595215e-01,8.17872881889343262e-01,1.18046358227729797e-01,-3.00728023052215576e-01,-4.70518797636032104e-01],"default_left":[false,false,false,false,false,false,false],"id":3,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[5.42109680175781250e+00,1.03034389019012451e+00,3.41049194335937500e+00,0.00000000000000000e+00,1.19803142547607422e+00,4.23731803894042969e+00,4.69757843017578125e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-2.07929229736328125e+00,-5.09094715118408203e-01,-8.72411578893661499e-02,1.22680939733982086e-01,1.77069548517465591e-02,-4.51092049479484558e-02,-7.05778226256370544e-02],"split_indices":[3,0,3,0,0,0,0],"sum_hessian":[5.04713470458984375e+02,9.86623668670654297e+00,4.94847229003906250e+02,2.13924217224121094e+00,7.72699451446533203e+00,2.30380615234375000e+02,2.64466613769531250e+02],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"0.500000","num_class":"0","num_feature":"4"},"objective":{"name":"reg:squarederror","reg_loss_param":{"scale_pos_weight":"1"}}},"version":[1,0,0]} \ No newline at end of file diff --git a/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.cls.bin b/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.cls.bin new file mode 100644 index 0000000000000000000000000000000000000000..342dde2a919950eaa0c01806afc6ce9044859dc3 GIT binary patch literal 5641 zcmeHLeQXp(6kqxw7LXtb5=zp$oDmak6lgW^xOT3)0kP6*gPH>3bnb3@x3C|V-MuTN zU~7aBF_Q9O!ymR+ia{j=P_$rLpxBmx6fu6J3Z$Vb)`)xrP0&`U^QL$0UXR2-{G)B? zOD1#k=FNL=-}}9{v&V59O5r#znT`o`j1*0c8dgHGDLDEnOJ27H@n5rVjmH1$&gzmn*HxvzSGi%wOL@0fq35BhWIl1rr{7V+G3Q(LHf6`Tix z{R1!m(+oNWI0L*B=A-5pk`*>~v=ql?c1nJ=+BO<|qfzr`%<} zDeG4zU)EKJBD*f|e`WUM!1=2Fy=c+Ojr>-}(qkas`Q|Rvvk3858rH%yFuHZ)UiYN$ zla9k;8vYQ?T;9pg_&bF?5BNHs4~`p6`2dW?D2Bh-dL(ARDBuL}1~jwai55D=;rXqK zTRE^7s~M0_+xzhG;kcjHu+=pj_wG~K&5PeXjsW*J%Qmz60bO##PL%H|MDTl??=SXAxadP#&k0kVKT|DI6)HRFs>jgc# zS-<_GIN9A;G(}kQdI#&zY1c2HtGD0dZ~J@L`GUw|cirb3`NJ6(o2gkYobi4J0t8q zQ(Uwitv>XU;69VW?k|Dv_mSuy_!z1zm@eP%SqakRfbi#=Qmh1+1ihMxA-gmmBy zet!ZU^Yb$k`Zc~9O!)pxy4sy^pILO}7<`k)_3Ng2_03YzYgnjg3WthC!;}hqCUiczs0a3ngPv6nT~*Dm|($A6om{)DUyo4 z0ZI1ZGEwpeOv?^)MK%ehdV@oJBjhoN5|l0Ocr|%|g+1k9c1YN|Ba2~CzVnu<`7DQY z(T|B73|QO*mP{0n&%!gjL>6&SGZI|)5JSX9nr;%+D`K-e2tLy+Nh-@$FoY$@U(}LUCWavzm~vLk&9u+Uw9j7&4}&cm uS(HJMNJT0f7WSY#JQpq%i5b$VCa)G!A?w+c8?#>gG%tW5sZ^qfTK_MSFT<$- literal 0 HcmV?d00001 diff --git a/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.cls.json b/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.cls.json new file mode 100644 index 000000000..495226599 --- /dev/null +++ b/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.cls.json @@ -0,0 +1 @@ +{"learner":{"attributes":{"scikit_learn":"{\"n_estimators\": 2, \"objective\": \"multi:softprob\", \"max_depth\": 2, \"learning_rate\": null, \"verbosity\": null, \"booster\": null, \"tree_method\": \"hist\", \"gamma\": null, \"min_child_weight\": null, \"max_delta_step\": null, \"subsample\": null, \"colsample_bytree\": null, \"colsample_bylevel\": null, \"colsample_bynode\": null, \"reg_alpha\": null, \"reg_lambda\": null, \"scale_pos_weight\": null, \"base_score\": null, \"missing\": NaN, \"num_parallel_tree\": 2, \"kwargs\": {}, \"random_state\": null, \"n_jobs\": null, \"monotone_constraints\": null, \"interaction_constraints\": null, \"importance_type\": \"gain\", \"gpu_id\": null, \"classes_\": [0, 1, 2], \"n_classes_\": 3, \"_le\": {\"classes_\": [0, 1, 2]}, \"_features_count\": 4, \"type\": \"XGBClassifier\"}"},"gradient_booster":{"model":{"gbtree_model_param":{"num_trees":"12","size_leaf_vector":"0"},"tree_info":[0,0,1,1,2,2,0,0,1,1,2,2],"trees":[{"base_weights":[-3.10496310703456402e-03,-1.14356480538845062e-01,1.12878747284412384e-01,-2.06186547875404358e-01,1.52976317331194878e-02,-5.11124253273010254e-01,1.29867866635322571e-01],"default_left":[false,false,false,false,false,false,false],"id":0,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.92025685310363770e+00,1.37833094596862793e+00,1.24116408824920654e+00,1.33592224121093750e+00,1.26208007335662842e+00,0.00000000000000000e+00,1.38571357727050781e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[1.14481402561068535e-02,2.25720971822738647e-01,-2.23139905929565430e+00,-3.09279840439558029e-02,2.29464494623243809e-03,-7.66686424612998962e-02,1.94801799952983856e-02],"split_indices":[1,0,0,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.14501411437988281e+02,1.09815666198730469e+02,6.66821746826171875e+01,4.78192405700683594e+01,2.13970732688903809e+00,1.07675956726074219e+02],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.10496310703456402e-03,-1.14356480538845062e-01,1.12878747284412384e-01,-2.06186547875404358e-01,1.52976317331194878e-02,-5.11124253273010254e-01,1.29867866635322571e-01],"default_left":[false,false,false,false,false,false,false],"id":1,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.92025685310363770e+00,1.37833094596862793e+00,1.24116408824920654e+00,1.33592224121093750e+00,1.26208007335662842e+00,0.00000000000000000e+00,1.38571357727050781e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[1.14481402561068535e-02,2.25720971822738647e-01,-2.23139905929565430e+00,-3.09279840439558029e-02,2.29464494623243809e-03,-7.66686424612998962e-02,1.94801799952983856e-02],"split_indices":[1,0,0,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.14501411437988281e+02,1.09815666198730469e+02,6.66821746826171875e+01,4.78192405700683594e+01,2.13970732688903809e+00,1.07675956726074219e+02],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-1.69028975069522858e-02,8.05607438087463379e-01,-2.48735304921865463e-02,1.18516474962234497e-01,-7.35604763031005859e-02],"default_left":[false,false,false,false,false],"id":2,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.49720096588134766e+00,0.00000000000000000e+00,1.57338166236877441e+00,1.47135305404663086e+00,1.12515950202941895e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.20841123163700104e-01,-6.54938280582427979e-01,1.77774727344512939e-02,-1.10340714454650879e-02],"split_indices":[0,0,1,0,0],"sum_hessian":[2.24317077636718750e+02,1.19245672225952148e+00,2.23124618530273438e+02,5.61932411193847656e+01,1.66931381225585938e+02],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[-1.69028975069522858e-02,8.05607438087463379e-01,-2.48735304921865463e-02,1.18516474962234497e-01,-7.35604763031005859e-02],"default_left":[false,false,false,false,false],"id":3,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.49720096588134766e+00,0.00000000000000000e+00,1.57338166236877441e+00,1.47135305404663086e+00,1.12515950202941895e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.20841123163700104e-01,-6.54938280582427979e-01,1.77774727344512939e-02,-1.10340714454650879e-02],"split_indices":[0,0,1,0,0],"sum_hessian":[2.24317077636718750e+02,1.19245672225952148e+00,2.23124618530273438e+02,5.61932411193847656e+01,1.66931381225585938e+02],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[2.00077202171087265e-02,9.98598039150238037e-02,-1.12991318106651306e-01,-2.68701147288084030e-02,2.14132413268089294e-01,-2.78286129236221313e-01,9.12456586956977844e-03],"default_left":[false,false,false,false,false,false,false],"id":4,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.40420699119567871e+00,2.05230426788330078e+00,1.71823227405548096e+00,3.13472294807434082e+00,2.89248418807983398e+00,9.98327255249023438e-01,1.37142074108123779e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[3.12443733215332031e-01,-5.36825239658355713e-01,6.99698746204376221e-01,-4.03051730245351791e-03,3.21198627352714539e-02,-4.17429208755493164e-02,1.36868492700159550e-03],"split_indices":[1,1,1,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.40319183349609375e+02,8.39979019165039062e+01,6.68957366943359375e+01,7.34234390258789062e+01,3.51459007263183594e+01,4.88520011901855469e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[2.00077202171087265e-02,9.98598039150238037e-02,-1.12991318106651306e-01,-2.68701147288084030e-02,2.14132413268089294e-01,-2.78286129236221313e-01,9.12456586956977844e-03],"default_left":[false,false,false,false,false,false,false],"id":5,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.40420699119567871e+00,2.05230426788330078e+00,1.71823227405548096e+00,3.13472294807434082e+00,2.89248418807983398e+00,9.98327255249023438e-01,1.37142074108123779e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[3.12443733215332031e-01,-5.36825239658355713e-01,6.99698746204376221e-01,-4.03051730245351791e-03,3.21198627352714539e-02,-4.17429208755493164e-02,1.36868492700159550e-03],"split_indices":[1,1,1,0,0,0,0],"sum_hessian":[2.24317077636718750e+02,1.40319183349609375e+02,8.39979019165039062e+01,6.68957366943359375e+01,7.34234390258789062e+01,3.51459007263183594e+01,4.88520011901855469e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.14401555806398392e-03,-2.77371466159820557e-01,2.71770134568214417e-02,-5.22169768810272217e-01,-1.32588371634483337e-01,-1.92690286785364151e-02,1.96981996297836304e-01],"default_left":[false,false,false,false,false,false,false],"id":6,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.87940013408660889e+00,7.28317379951477051e-01,1.60969936847686768e+00,3.83746862411499023e-01,1.40333545207977295e+00,1.32722187042236328e+00,1.38107287883758545e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-1.36123311519622803e+00,-3.55613559484481812e-01,8.09594929218292236e-01,-7.83254653215408325e-02,-1.98882557451725006e-02,-2.89035448804497719e-03,2.95473001897335052e-02],"split_indices":[1,2,0,0,0,0,0],"sum_hessian":[2.24122543334960938e+02,2.15025730133056641e+01,2.02619964599609375e+02,7.02246618270874023e+00,1.44801063537597656e+01,1.59797698974609375e+02,4.28222694396972656e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-3.14401555806398392e-03,-2.77371466159820557e-01,2.71770134568214417e-02,-5.22169768810272217e-01,-1.32588371634483337e-01,-1.92690286785364151e-02,1.96981996297836304e-01],"default_left":[false,false,false,false,false,false,false],"id":7,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.87940013408660889e+00,7.28317379951477051e-01,1.60969936847686768e+00,3.83746862411499023e-01,1.40333545207977295e+00,1.32722187042236328e+00,1.38107287883758545e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-1.36123311519622803e+00,-3.55613559484481812e-01,8.09594929218292236e-01,-7.83254653215408325e-02,-1.98882557451725006e-02,-2.89035448804497719e-03,2.95473001897335052e-02],"split_indices":[1,2,0,0,0,0,0],"sum_hessian":[2.24122543334960938e+02,2.15025730133056641e+01,2.02619964599609375e+02,7.02246618270874023e+00,1.44801063537597656e+01,1.59797698974609375e+02,4.28222694396972656e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[-1.22269820421934128e-02,7.03054547309875488e-01,-1.94818396121263504e-02,-3.24203968048095703e-02,3.11440438032150269e-01],"default_left":[false,false,false,false,false],"id":8,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.17855262756347656e+00,0.00000000000000000e+00,9.65734302997589111e-01,1.15091991424560547e+00,1.31937885284423828e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.05458185076713562e-01,1.78177630901336670e+00,-4.86305961385369301e-03,4.67160679399967194e-02],"split_indices":[0,0,2,0,0],"sum_hessian":[2.23474166870117188e+02,1.28086733818054199e+00,2.22193298339843750e+02,2.14700851440429688e+02,7.49244642257690430e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[-1.22269820421934128e-02,7.03054547309875488e-01,-1.94818396121263504e-02,-3.24203968048095703e-02,3.11440438032150269e-01],"default_left":[false,false,false,false,false],"id":9,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[1.17855262756347656e+00,0.00000000000000000e+00,9.65734302997589111e-01,1.15091991424560547e+00,1.31937885284423828e+00],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-2.83283019065856934e+00,1.05458185076713562e-01,1.78177630901336670e+00,-4.86305961385369301e-03,4.67160679399967194e-02],"split_indices":[0,0,2,0,0],"sum_hessian":[2.23474166870117188e+02,1.28086733818054199e+00,2.22193298339843750e+02,2.14700851440429688e+02,7.49244642257690430e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[1.52873406186699867e-02,5.03631830215454102e-01,-6.68718665838241577e-03,-9.80335399508476257e-02,7.35435545444488525e-01,3.88425774872303009e-02,-1.30366414785385132e-01],"default_left":[false,false,false,false,false,false,false],"id":10,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.42684388160705566e+00,1.43019437789916992e+00,1.22756230831146240e+00,3.81559073925018311e-01,3.57241153717041016e-01,1.24453198909759521e+00,1.17859208583831787e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-2.00600242614746094e+00,-2.40636110305786133e+00,6.73008561134338379e-01,-1.47050311788916588e-02,1.10315337777137756e-01,5.82638662308454514e-03,-1.95549633353948593e-02],"split_indices":[1,1,2,0,0,0,0],"sum_hessian":[2.24835845947265625e+02,8.73767948150634766e+00,2.16098175048828125e+02,2.59061360359191895e+00,6.14706563949584961e+00,1.58453109741210938e+02,5.76450576782226562e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[1.52873406186699867e-02,5.03631830215454102e-01,-6.68718665838241577e-03,-9.80335399508476257e-02,7.35435545444488525e-01,3.88425774872303009e-02,-1.30366414785385132e-01],"default_left":[false,false,false,false,false,false,false],"id":11,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[2.42684388160705566e+00,1.43019437789916992e+00,1.22756230831146240e+00,3.81559073925018311e-01,3.57241153717041016e-01,1.24453198909759521e+00,1.17859208583831787e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[-2.00600242614746094e+00,-2.40636110305786133e+00,6.73008561134338379e-01,-1.47050311788916588e-02,1.10315337777137756e-01,5.82638662308454514e-03,-1.95549633353948593e-02],"split_indices":[1,1,2,0,0,0,0],"sum_hessian":[2.24835845947265625e+02,8.73767948150634766e+00,2.16098175048828125e+02,2.59061360359191895e+00,6.14706563949584961e+00,1.58453109741210938e+02,5.76450576782226562e+01],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"0.500000","num_class":"3","num_feature":"4"},"objective":{"name":"multi:softprob","softmax_multiclass_param":{"num_class":"3"}}},"version":[1,0,0]} \ No newline at end of file diff --git a/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.ltr.bin b/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.ltr.bin new file mode 100644 index 0000000000000000000000000000000000000000..8d7ff40116e1da353faeff931f7c9bcef65e154e GIT binary patch literal 2445 zcmeHJO-L0{6rN0relUWFf+TxTi-^JsBGH-mf{HdSTD7QdcV=$$uHLzKn0wz-5kW}% zZzBjr5C%oRiy)y;pB4nIY9U0pX{8lNw5S%fv3ust^Nl97iFDzC;mkefJ3pN7%#UFh zR;5vtWwBlewJI(eV%}2EUB_Dy8$4Tugg}c6+1}qq0=-owK&6o;O#l6MS_|}5p|9oo zdU9)URI4tqgXnwXRHs(}4gS)MA0Iw#>6lr#@6RqZK%Qb`F*ft^%dkBE-t5~skQqO) zv983=kDFsw+vgV}kav7{!F*@j%be`5Z-n^Q8#Y@{fA(aiE=)Fp|4c=Vb$Ls7dd;>Y z5q@co|89;eVZgC0wtWPi^CRmmt-6xC*WWt|I~qMpr8mGt<--w;pS6$UJFXNUM{*z2 zZ^nk^?(Or2-r>1>+f#j`@$K1e>;A-OM$~MrgL>@xq*-yhFVj5qu>km2*M6CO)eq8N zCk9M7gUAo-(sb%mw5P|=RdfH{bD7nT_C(JQ_{H`+*2TeZGv|X`pB~M3=Wh)(4n+U} literal 0 HcmV?d00001 diff --git a/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.ltr.json b/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.ltr.json new file mode 100644 index 000000000..312768898 --- /dev/null +++ b/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.ltr.json @@ -0,0 +1 @@ +{"learner":{"attributes":{"scikit_learn":"{\"n_estimators\": 2, \"objective\": \"rank:ndcg\", \"max_depth\": 2, \"learning_rate\": null, \"verbosity\": null, \"booster\": null, \"tree_method\": \"hist\", \"gamma\": null, \"min_child_weight\": null, \"max_delta_step\": null, \"subsample\": null, \"colsample_bytree\": null, \"colsample_bylevel\": null, \"colsample_bynode\": null, \"reg_alpha\": null, \"reg_lambda\": null, \"scale_pos_weight\": null, \"base_score\": null, \"missing\": NaN, \"num_parallel_tree\": 2, \"kwargs\": {}, \"random_state\": null, \"n_jobs\": null, \"monotone_constraints\": null, \"interaction_constraints\": null, \"importance_type\": \"gain\", \"gpu_id\": null, \"type\": \"XGBRanker\"}"},"gradient_booster":{"model":{"gbtree_model_param":{"num_trees":"4","size_leaf_vector":"0"},"tree_info":[0,0,0,0],"trees":[{"base_weights":[2.18596185597164094e-09,-3.76773595809936523e-01,4.55630868673324585e-02,1.12075649201869965e-01,-1.93485423922538757e-01],"default_left":[false,false,false,false,false],"id":0,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[4.20947641134262085e-01,0.00000000000000000e+00,3.69498044252395630e-01,5.97973287105560303e-01,6.13317489624023438e-01],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-1.45796775817871094e+00,-5.65160401165485382e-02,8.68250608444213867e-01,1.68113484978675842e-02,-2.90228147059679031e-02],"split_indices":[3,0,1,0,0],"sum_hessian":[2.25207920074462891e+01,1.64538443088531494e+00,2.08754062652587891e+01,1.67469234466552734e+01,4.12848377227783203e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[2.18596185597164094e-09,-3.76773595809936523e-01,4.55630868673324585e-02,1.12075649201869965e-01,-1.93485423922538757e-01],"default_left":[false,false,false,false,false],"id":1,"leaf_child_counts":[1,0,2,0,0],"left_children":[1,-1,3,-1,-1],"loss_changes":[4.20947641134262085e-01,0.00000000000000000e+00,3.69498044252395630e-01,5.97973287105560303e-01,6.13317489624023438e-01],"parents":[2147483647,0,0,2,2],"right_children":[2,-1,4,-1,-1],"split_conditions":[-1.45796775817871094e+00,-5.65160401165485382e-02,8.68250608444213867e-01,1.68113484978675842e-02,-2.90228147059679031e-02],"split_indices":[3,0,1,0,0],"sum_hessian":[2.25207920074462891e+01,1.64538443088531494e+00,2.08754062652587891e+01,1.67469234466552734e+01,4.12848377227783203e+00],"tree_param":{"num_feature":"4","num_nodes":"5","size_leaf_vector":"0"}},{"base_weights":[2.31542762740843955e-09,-1.12662151455879211e-01,3.53309124708175659e-01,-4.52967911958694458e-01,-4.28877249360084534e-02,-1.19008123874664307e-01,4.98231500387191772e-01],"default_left":[false,false,false,false,false,false,false],"id":2,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.03438735008239746e+00,4.48428511619567871e-01,4.89362835884094238e-01,0.00000000000000000e+00,2.74164468050003052e-01,0.00000000000000000e+00,0.00000000000000000e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[5.69312453269958496e-01,-1.49666213989257812e+00,-3.32068562507629395e-01,-6.79451897740364075e-02,-6.43315911293029785e-03,-1.78512185811996460e-02,7.47347250580787659e-02],"split_indices":[1,1,0,0,0,0,0],"sum_hessian":[2.39866485595703125e+01,1.87036170959472656e+01,5.28303003311157227e+00,2.24795222282409668e+00,1.64556655883789062e+01,1.28239238262176514e+00,4.00063753128051758e+00],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}},{"base_weights":[2.31542762740843955e-09,-1.12662151455879211e-01,3.53309124708175659e-01,-4.52967911958694458e-01,-4.28877249360084534e-02,-1.19008123874664307e-01,4.98231500387191772e-01],"default_left":[false,false,false,false,false,false,false],"id":3,"leaf_child_counts":[0,2,2,0,0,0,0],"left_children":[1,3,5,-1,-1,-1,-1],"loss_changes":[1.03438735008239746e+00,4.48428511619567871e-01,4.89362835884094238e-01,0.00000000000000000e+00,2.74164468050003052e-01,0.00000000000000000e+00,0.00000000000000000e+00],"parents":[2147483647,0,0,1,1,2,2],"right_children":[2,4,6,-1,-1,-1,-1],"split_conditions":[5.69312453269958496e-01,-1.49666213989257812e+00,-3.32068562507629395e-01,-6.79451897740364075e-02,-6.43315911293029785e-03,-1.78512185811996460e-02,7.47347250580787659e-02],"split_indices":[1,1,0,0,0,0,0],"sum_hessian":[2.39866485595703125e+01,1.87036170959472656e+01,5.28303003311157227e+00,2.24795222282409668e+00,1.64556655883789062e+01,1.28239238262176514e+00,4.00063753128051758e+00],"tree_param":{"num_feature":"4","num_nodes":"7","size_leaf_vector":"0"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"0.500000","num_class":"0","num_feature":"4"},"objective":{"lambda_rank_param":{"fix_list_weight":"0","num_pairsample":"1"},"name":"rank:ndcg"}},"version":[1,0,0]} \ No newline at end of file diff --git a/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.reg.bin b/tests/python/models/xgboost_scikit-1.0.0-SNAPSHOT.reg.bin new file mode 100644 index 0000000000000000000000000000000000000000..1be34798dbb26e86dd5516d92ad321c7e1cd933b GIT binary patch literal 2606 zcmeHIT}TvB6rO(kDf**c0xOnJdk|?NqW_Q`~2YSmOFU*`d-}&x2XU@52 zM$@$N#aenT$jG@6t1>}3iIs{o3Cu~9aL#ys29KGN69SpeWtM-c4CGc2AZ05Xfd2a9 zFSQ`Ah3W#iD^*#hL~4<`W%GM7GS;tDEl*w9ijO}k&=o%g`(c3{59od1BaNi7m8EBU zmG3FMVmb`i6yC9+D-rG8YyIfSpT?G3OZ<4D%|m+gSf^2P?_(eEb?d7BVC1+_*H_#R zy7s=I%o$x;F?h4b=T8j0(j#3R#>4Z2ePA9O@6h)zuQ8I_clh>Cn^x;LigxOk3%{UNuDs~O`ii#p+hs5hseN@ir<2B^;g|lr zC`haQQmAh@b)+1g*PiGuzi3SA2cLh7fWGef1>M7^jr{N3zQ4}uYx?6o?Z(^D315d? z|K<7k-}(VP3ohsntYvAT3CnJXE(Vl8rY-?gD~~tfiwZ4*fFC4Ojwu_9RcRc%QRb)$Oew#M>X`|gOgkLrGRa(mEanQyvF*x$ zRcVA{)pn*zJuAWS7?;?!2$D$;a#mfkUB6pi0@={NOr)&6_CRGdiE$zX