├── .bazelrc ├── .clang-format ├── .gitignore ├── LICENSE ├── README.md ├── WORKSPACE ├── docs ├── ALGORITHMS.md ├── INSTALL.md ├── PERFORMANCE_BENCHMARK.md ├── TUTORIAL.md └── images │ ├── feature_importance.png │ ├── partial_dependency_plot.png │ └── partial_dependency_plot2.png ├── examples ├── adult │ ├── README.md │ ├── example.ipynb │ ├── test.tsv │ ├── train.tsv │ └── univariate_example.ipynb └── benchm-ml │ ├── benchm-ml.auc.config │ ├── benchm-ml.gbrank.config │ ├── benchm-ml.huberized_hinge.config │ ├── benchm-ml.logloss.config │ ├── benchm-ml.mse.config │ ├── benchm-ml.pairwise_logloss.config │ ├── python_example.py │ ├── run.py │ ├── test.tsv │ └── train-0.1m.tsv ├── gbdt ├── __init__.py ├── _data_store.py ├── _forest.py ├── _forest_visualizer.py ├── _gbdt.py ├── _libgbdt.py ├── _partial_dependency_plot.py ├── _version.py └── lib │ ├── __init__.py │ ├── darwin_x86_64 │ ├── __init__.py │ ├── py27 │ │ ├── __init__.py │ │ └── libgbdt.so │ └── py3 │ │ ├── __init__.py │ │ └── libgbdt.so │ └── linux_x86_64 │ ├── __init__.py │ └── py27 │ ├── __init__.py │ └── libgbdt.so ├── scripts ├── bagged.py ├── compute_metrics.py ├── compute_rmse.py ├── convert_tsv_to_filatfiles.py └── draw_trees.py ├── setup.cfg ├── setup.py ├── setup_dependencies └── setup.sh ├── src ├── BUILD ├── base │ ├── BUILD │ └── base.h ├── data_store │ ├── BUILD │ ├── column.cc │ ├── column.h │ ├── column_test.cc │ ├── data_store.cc │ ├── data_store.h │ ├── data_store_test.cc │ ├── flatfiles_data_store.cc │ ├── flatfiles_data_store.h │ ├── flatfiles_data_store_test.cc │ ├── testdata │ │ ├── flatfiles_data_store_test │ │ │ ├── bar │ │ │ ├── foo │ │ │ └── foo2 │ │ └── tsv_data_store_test │ │ │ ├── block-0-with-header.tsv │ │ │ ├── block-0.tsv │ │ │ ├── block-1.tsv │ │ │ ├── block-2.tsv │ │ │ ├── data_with_missing.tsv │ │ │ └── header │ ├── tsv_block.cc │ ├── tsv_block.h │ ├── tsv_block_test.cc │ ├── tsv_data_store.cc │ ├── tsv_data_store.h │ └── tsv_data_store_test.cc ├── flags.cc ├── gbdt_algo │ ├── BUILD │ ├── compute_tree_scores.cc │ ├── compute_tree_scores.h │ ├── compute_tree_scores_test.cc │ ├── evaluation.cc │ ├── evaluation.h │ ├── gbdt_algo.cc │ ├── gbdt_algo.h │ ├── gbdt_algo_test.cc │ ├── split_algo.cc │ ├── split_algo.h │ ├── split_algo_test.cc │ ├── testdata │ │ └── gbdt_algo_test │ │ │ ├── flatfiles │ │ │ ├── color │ │ │ ├── height │ │ │ ├── label │ │ │ └── width │ │ │ └── forest.model.txt │ ├── tree_algo.cc │ ├── tree_algo.h │ ├── tree_algo_test.cc │ ├── utils.cc │ ├── utils.h │ └── utils_test.cc ├── loss_func │ ├── BUILD │ ├── gradient_data.h │ ├── group.cc │ ├── group.h │ ├── group_test.cc │ ├── loss_func.h │ ├── loss_func_auc.h │ ├── loss_func_factory.cc │ ├── loss_func_factory.h │ ├── loss_func_factory_test.cc │ ├── loss_func_gbrank.h │ ├── loss_func_huberized_hinge.cc │ ├── loss_func_huberized_hinge.h │ ├── loss_func_huberized_hinge_test.cc │ ├── loss_func_lambdamart.cc │ ├── loss_func_lambdamart.h │ ├── loss_func_lambdamart_test.cc │ ├── loss_func_logloss.cc │ ├── loss_func_logloss.h │ ├── loss_func_logloss_test.cc │ ├── loss_func_math.cc │ ├── loss_func_math.h │ ├── loss_func_mse.h │ ├── loss_func_mse_test.cc │ ├── loss_func_pairwise.cc │ ├── loss_func_pairwise.h │ ├── loss_func_pairwise_logloss.h │ ├── loss_func_pairwise_test.cc │ ├── loss_func_pointwise.cc │ └── loss_func_pointwise.h ├── main.cc ├── proto │ ├── BUILD │ ├── config.proto │ └── tree.proto ├── python │ ├── BUILD │ ├── column_py.cc │ ├── column_py.h │ ├── datastore_py.cc │ ├── datastore_py.h │ ├── forest_py.cc │ ├── forest_py.h │ ├── gbdt_py.cc │ ├── gbdt_py_base.cc │ ├── gbdt_py_base.h │ ├── train_gbdt_py.cc │ └── train_gbdt_py.h └── utils │ ├── BUILD │ ├── json_utils.cc │ ├── json_utils.h │ ├── json_utils_test.cc │ ├── stopwatch.cc │ ├── stopwatch.h │ ├── stopwatch_test.cc │ ├── subsampling.cc │ ├── subsampling.h │ ├── subsampling_test.cc │ ├── threadpool.cc │ ├── threadpool.h │ ├── utils.cc │ ├── utils.h │ ├── utils_inl.h │ ├── utils_test.cc │ ├── vector_slice.h │ └── vector_slice_test.cc └── third_party ├── BUILD ├── cppformat.BUILD ├── py ├── BUILD ├── BUILD.tpl ├── python_configure.bzl └── remote.BUILD.tpl └── pybind11.BUILD /.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/.bazelrc -------------------------------------------------------------------------------- /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/.clang-format -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/README.md -------------------------------------------------------------------------------- /WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/WORKSPACE -------------------------------------------------------------------------------- /docs/ALGORITHMS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/docs/ALGORITHMS.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /docs/PERFORMANCE_BENCHMARK.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/docs/PERFORMANCE_BENCHMARK.md -------------------------------------------------------------------------------- /docs/TUTORIAL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/docs/TUTORIAL.md -------------------------------------------------------------------------------- /docs/images/feature_importance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/docs/images/feature_importance.png -------------------------------------------------------------------------------- /docs/images/partial_dependency_plot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/docs/images/partial_dependency_plot.png -------------------------------------------------------------------------------- /docs/images/partial_dependency_plot2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/docs/images/partial_dependency_plot2.png -------------------------------------------------------------------------------- /examples/adult/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/adult/README.md -------------------------------------------------------------------------------- /examples/adult/example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/adult/example.ipynb -------------------------------------------------------------------------------- /examples/adult/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/adult/test.tsv -------------------------------------------------------------------------------- /examples/adult/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/adult/train.tsv -------------------------------------------------------------------------------- /examples/adult/univariate_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/adult/univariate_example.ipynb -------------------------------------------------------------------------------- /examples/benchm-ml/benchm-ml.auc.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/benchm-ml.auc.config -------------------------------------------------------------------------------- /examples/benchm-ml/benchm-ml.gbrank.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/benchm-ml.gbrank.config -------------------------------------------------------------------------------- /examples/benchm-ml/benchm-ml.huberized_hinge.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/benchm-ml.huberized_hinge.config -------------------------------------------------------------------------------- /examples/benchm-ml/benchm-ml.logloss.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/benchm-ml.logloss.config -------------------------------------------------------------------------------- /examples/benchm-ml/benchm-ml.mse.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/benchm-ml.mse.config -------------------------------------------------------------------------------- /examples/benchm-ml/benchm-ml.pairwise_logloss.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/benchm-ml.pairwise_logloss.config -------------------------------------------------------------------------------- /examples/benchm-ml/python_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/python_example.py -------------------------------------------------------------------------------- /examples/benchm-ml/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/run.py -------------------------------------------------------------------------------- /examples/benchm-ml/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/test.tsv -------------------------------------------------------------------------------- /examples/benchm-ml/train-0.1m.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/examples/benchm-ml/train-0.1m.tsv -------------------------------------------------------------------------------- /gbdt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/__init__.py -------------------------------------------------------------------------------- /gbdt/_data_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/_data_store.py -------------------------------------------------------------------------------- /gbdt/_forest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/_forest.py -------------------------------------------------------------------------------- /gbdt/_forest_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/_forest_visualizer.py -------------------------------------------------------------------------------- /gbdt/_gbdt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/_gbdt.py -------------------------------------------------------------------------------- /gbdt/_libgbdt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/_libgbdt.py -------------------------------------------------------------------------------- /gbdt/_partial_dependency_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/_partial_dependency_plot.py -------------------------------------------------------------------------------- /gbdt/_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/_version.py -------------------------------------------------------------------------------- /gbdt/lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gbdt/lib/darwin_x86_64/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gbdt/lib/darwin_x86_64/py27/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gbdt/lib/darwin_x86_64/py27/libgbdt.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/lib/darwin_x86_64/py27/libgbdt.so -------------------------------------------------------------------------------- /gbdt/lib/darwin_x86_64/py3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gbdt/lib/darwin_x86_64/py3/libgbdt.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/lib/darwin_x86_64/py3/libgbdt.so -------------------------------------------------------------------------------- /gbdt/lib/linux_x86_64/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gbdt/lib/linux_x86_64/py27/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gbdt/lib/linux_x86_64/py27/libgbdt.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/gbdt/lib/linux_x86_64/py27/libgbdt.so -------------------------------------------------------------------------------- /scripts/bagged.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/scripts/bagged.py -------------------------------------------------------------------------------- /scripts/compute_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/scripts/compute_metrics.py -------------------------------------------------------------------------------- /scripts/compute_rmse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/scripts/compute_rmse.py -------------------------------------------------------------------------------- /scripts/convert_tsv_to_filatfiles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/scripts/convert_tsv_to_filatfiles.py -------------------------------------------------------------------------------- /scripts/draw_trees.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/scripts/draw_trees.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [bdist_wheel] -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/setup.py -------------------------------------------------------------------------------- /setup_dependencies/setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/setup_dependencies/setup.sh -------------------------------------------------------------------------------- /src/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/BUILD -------------------------------------------------------------------------------- /src/base/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/base/BUILD -------------------------------------------------------------------------------- /src/base/base.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/base/base.h -------------------------------------------------------------------------------- /src/data_store/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/BUILD -------------------------------------------------------------------------------- /src/data_store/column.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/column.cc -------------------------------------------------------------------------------- /src/data_store/column.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/column.h -------------------------------------------------------------------------------- /src/data_store/column_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/column_test.cc -------------------------------------------------------------------------------- /src/data_store/data_store.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/data_store.cc -------------------------------------------------------------------------------- /src/data_store/data_store.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/data_store.h -------------------------------------------------------------------------------- /src/data_store/data_store_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/data_store_test.cc -------------------------------------------------------------------------------- /src/data_store/flatfiles_data_store.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/flatfiles_data_store.cc -------------------------------------------------------------------------------- /src/data_store/flatfiles_data_store.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/flatfiles_data_store.h -------------------------------------------------------------------------------- /src/data_store/flatfiles_data_store_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/flatfiles_data_store_test.cc -------------------------------------------------------------------------------- /src/data_store/testdata/flatfiles_data_store_test/bar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/testdata/flatfiles_data_store_test/bar -------------------------------------------------------------------------------- /src/data_store/testdata/flatfiles_data_store_test/foo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/testdata/flatfiles_data_store_test/foo -------------------------------------------------------------------------------- /src/data_store/testdata/flatfiles_data_store_test/foo2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/testdata/flatfiles_data_store_test/foo2 -------------------------------------------------------------------------------- /src/data_store/testdata/tsv_data_store_test/block-0-with-header.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/testdata/tsv_data_store_test/block-0-with-header.tsv -------------------------------------------------------------------------------- /src/data_store/testdata/tsv_data_store_test/block-0.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/testdata/tsv_data_store_test/block-0.tsv -------------------------------------------------------------------------------- /src/data_store/testdata/tsv_data_store_test/block-1.tsv: -------------------------------------------------------------------------------- 1 | 2 blue 45 2.3 clear 2 | 3 red 672 4.5 snowy 3 | 1 yellow 123 6.5 rainy 4 | -------------------------------------------------------------------------------- /src/data_store/testdata/tsv_data_store_test/block-2.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/testdata/tsv_data_store_test/block-2.tsv -------------------------------------------------------------------------------- /src/data_store/testdata/tsv_data_store_test/data_with_missing.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/testdata/tsv_data_store_test/data_with_missing.tsv -------------------------------------------------------------------------------- /src/data_store/testdata/tsv_data_store_test/header: -------------------------------------------------------------------------------- 1 | target color foo bar weather 2 | -------------------------------------------------------------------------------- /src/data_store/tsv_block.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/tsv_block.cc -------------------------------------------------------------------------------- /src/data_store/tsv_block.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/tsv_block.h -------------------------------------------------------------------------------- /src/data_store/tsv_block_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/tsv_block_test.cc -------------------------------------------------------------------------------- /src/data_store/tsv_data_store.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/tsv_data_store.cc -------------------------------------------------------------------------------- /src/data_store/tsv_data_store.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/tsv_data_store.h -------------------------------------------------------------------------------- /src/data_store/tsv_data_store_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/data_store/tsv_data_store_test.cc -------------------------------------------------------------------------------- /src/flags.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/flags.cc -------------------------------------------------------------------------------- /src/gbdt_algo/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/BUILD -------------------------------------------------------------------------------- /src/gbdt_algo/compute_tree_scores.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/compute_tree_scores.cc -------------------------------------------------------------------------------- /src/gbdt_algo/compute_tree_scores.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/compute_tree_scores.h -------------------------------------------------------------------------------- /src/gbdt_algo/compute_tree_scores_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/compute_tree_scores_test.cc -------------------------------------------------------------------------------- /src/gbdt_algo/evaluation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/evaluation.cc -------------------------------------------------------------------------------- /src/gbdt_algo/evaluation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/evaluation.h -------------------------------------------------------------------------------- /src/gbdt_algo/gbdt_algo.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/gbdt_algo.cc -------------------------------------------------------------------------------- /src/gbdt_algo/gbdt_algo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/gbdt_algo.h -------------------------------------------------------------------------------- /src/gbdt_algo/gbdt_algo_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/gbdt_algo_test.cc -------------------------------------------------------------------------------- /src/gbdt_algo/split_algo.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/split_algo.cc -------------------------------------------------------------------------------- /src/gbdt_algo/split_algo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/split_algo.h -------------------------------------------------------------------------------- /src/gbdt_algo/split_algo_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/split_algo_test.cc -------------------------------------------------------------------------------- /src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/color: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/color -------------------------------------------------------------------------------- /src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/height: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/height -------------------------------------------------------------------------------- /src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/label: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/label -------------------------------------------------------------------------------- /src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/width: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/testdata/gbdt_algo_test/flatfiles/width -------------------------------------------------------------------------------- /src/gbdt_algo/testdata/gbdt_algo_test/forest.model.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/testdata/gbdt_algo_test/forest.model.txt -------------------------------------------------------------------------------- /src/gbdt_algo/tree_algo.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/tree_algo.cc -------------------------------------------------------------------------------- /src/gbdt_algo/tree_algo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/tree_algo.h -------------------------------------------------------------------------------- /src/gbdt_algo/tree_algo_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/tree_algo_test.cc -------------------------------------------------------------------------------- /src/gbdt_algo/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/utils.cc -------------------------------------------------------------------------------- /src/gbdt_algo/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/utils.h -------------------------------------------------------------------------------- /src/gbdt_algo/utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/gbdt_algo/utils_test.cc -------------------------------------------------------------------------------- /src/loss_func/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/BUILD -------------------------------------------------------------------------------- /src/loss_func/gradient_data.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/gradient_data.h -------------------------------------------------------------------------------- /src/loss_func/group.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/group.cc -------------------------------------------------------------------------------- /src/loss_func/group.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/group.h -------------------------------------------------------------------------------- /src/loss_func/group_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/group_test.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_auc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_auc.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_factory.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_factory.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_factory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_factory.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_factory_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_factory_test.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_gbrank.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_gbrank.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_huberized_hinge.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_huberized_hinge.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_huberized_hinge.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_huberized_hinge.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_huberized_hinge_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_huberized_hinge_test.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_lambdamart.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_lambdamart.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_lambdamart.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_lambdamart.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_lambdamart_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_lambdamart_test.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_logloss.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_logloss.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_logloss.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_logloss.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_logloss_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_logloss_test.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_math.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_math.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_math.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_math.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_mse.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_mse.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_mse_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_mse_test.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_pairwise.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_pairwise.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_pairwise.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_pairwise.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_pairwise_logloss.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_pairwise_logloss.h -------------------------------------------------------------------------------- /src/loss_func/loss_func_pairwise_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_pairwise_test.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_pointwise.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_pointwise.cc -------------------------------------------------------------------------------- /src/loss_func/loss_func_pointwise.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/loss_func/loss_func_pointwise.h -------------------------------------------------------------------------------- /src/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/main.cc -------------------------------------------------------------------------------- /src/proto/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/proto/BUILD -------------------------------------------------------------------------------- /src/proto/config.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/proto/config.proto -------------------------------------------------------------------------------- /src/proto/tree.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/proto/tree.proto -------------------------------------------------------------------------------- /src/python/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/BUILD -------------------------------------------------------------------------------- /src/python/column_py.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/column_py.cc -------------------------------------------------------------------------------- /src/python/column_py.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/column_py.h -------------------------------------------------------------------------------- /src/python/datastore_py.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/datastore_py.cc -------------------------------------------------------------------------------- /src/python/datastore_py.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/datastore_py.h -------------------------------------------------------------------------------- /src/python/forest_py.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/forest_py.cc -------------------------------------------------------------------------------- /src/python/forest_py.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/forest_py.h -------------------------------------------------------------------------------- /src/python/gbdt_py.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/gbdt_py.cc -------------------------------------------------------------------------------- /src/python/gbdt_py_base.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/gbdt_py_base.cc -------------------------------------------------------------------------------- /src/python/gbdt_py_base.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/gbdt_py_base.h -------------------------------------------------------------------------------- /src/python/train_gbdt_py.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/train_gbdt_py.cc -------------------------------------------------------------------------------- /src/python/train_gbdt_py.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/python/train_gbdt_py.h -------------------------------------------------------------------------------- /src/utils/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/BUILD -------------------------------------------------------------------------------- /src/utils/json_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/json_utils.cc -------------------------------------------------------------------------------- /src/utils/json_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/json_utils.h -------------------------------------------------------------------------------- /src/utils/json_utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/json_utils_test.cc -------------------------------------------------------------------------------- /src/utils/stopwatch.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/stopwatch.cc -------------------------------------------------------------------------------- /src/utils/stopwatch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/stopwatch.h -------------------------------------------------------------------------------- /src/utils/stopwatch_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/stopwatch_test.cc -------------------------------------------------------------------------------- /src/utils/subsampling.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/subsampling.cc -------------------------------------------------------------------------------- /src/utils/subsampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/subsampling.h -------------------------------------------------------------------------------- /src/utils/subsampling_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/subsampling_test.cc -------------------------------------------------------------------------------- /src/utils/threadpool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/threadpool.cc -------------------------------------------------------------------------------- /src/utils/threadpool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/threadpool.h -------------------------------------------------------------------------------- /src/utils/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/utils.cc -------------------------------------------------------------------------------- /src/utils/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/utils.h -------------------------------------------------------------------------------- /src/utils/utils_inl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/utils_inl.h -------------------------------------------------------------------------------- /src/utils/utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/utils_test.cc -------------------------------------------------------------------------------- /src/utils/vector_slice.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/vector_slice.h -------------------------------------------------------------------------------- /src/utils/vector_slice_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/src/utils/vector_slice_test.cc -------------------------------------------------------------------------------- /third_party/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/third_party/BUILD -------------------------------------------------------------------------------- /third_party/cppformat.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/third_party/cppformat.BUILD -------------------------------------------------------------------------------- /third_party/py/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/py/BUILD.tpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/third_party/py/BUILD.tpl -------------------------------------------------------------------------------- /third_party/py/python_configure.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/third_party/py/python_configure.bzl -------------------------------------------------------------------------------- /third_party/py/remote.BUILD.tpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/third_party/py/remote.BUILD.tpl -------------------------------------------------------------------------------- /third_party/pybind11.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yarny/gbdt/HEAD/third_party/pybind11.BUILD --------------------------------------------------------------------------------