{"id":18802400,"url":"https://github.com/jd-opensource/utboost","last_synced_at":"2025-04-13T18:10:44.047Z","repository":{"id":211363786,"uuid":"671794939","full_name":"jd-opensource/UTBoost","owner":"jd-opensource","description":"A powerful tree-based uplift modeling system.","archived":false,"fork":false,"pushed_at":"2024-01-23T07:27:35.000Z","size":193,"stargazers_count":30,"open_issues_count":0,"forks_count":2,"subscribers_count":4,"default_branch":"master","last_synced_at":"2025-04-12T23:07:20.968Z","etag":null,"topics":["causal-inference","descision-tree","gbm","gradient-boosting","uplift-modeling"],"latest_commit_sha":null,"homepage":"","language":"C++","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/jd-opensource.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2023-07-28T06:43:04.000Z","updated_at":"2025-04-02T06:44:01.000Z","dependencies_parsed_at":"2023-12-08T03:28:18.399Z","dependency_job_id":"af52f764-482c-462f-becd-148a73f66d1d","html_url":"https://github.com/jd-opensource/UTBoost","commit_stats":null,"previous_names":["jd-opensource/utboost"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/jd-opensource%2FUTBoost","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/jd-opensource%2FUTBoost/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/jd-opensource%2FUTBoost/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/jd-opensource%2FUTBoost/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/jd-opensource","download_url":"https://codeload.github.com/jd-opensource/UTBoost/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":248643004,"owners_count":21138355,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["causal-inference","descision-tree","gbm","gradient-boosting","uplift-modeling"],"created_at":"2024-11-07T22:27:56.428Z","updated_at":"2025-04-13T18:10:44.026Z","avatar_url":"https://github.com/jd-opensource.png","language":"C++","readme":"# UTBoost\r\n\r\nUTBoost is a powerful uplift modeling library based on boosting framework over decision trees.\r\nIt can handle large-scale RCT (randomized controlled trial) datasets and demonstrates superior predictive performance.\r\n\r\n## Documentations\r\n\r\n- [Installation Instructions](./doc/install.md)\r\n- [Applying Model](./doc/applying_models.md)\r\n\r\n## Quick Start\r\n\r\nSee the **[tutorial notebook](./examples/tutorial.ipynb)** for details.\r\n\r\n```python\r\n# import approaches\r\nfrom utboost import UTBClassifier, UTBRegressor\r\n\r\n# define model (CausalGBM algorithm)\r\nmodel = UTBClassifier(\r\n    ensemble_type='boosting',\r\n    criterion='gbm',\r\n    iterations=20,\r\n    max_depth=4\r\n)\r\n\r\n# fit model\r\nmodel.fit(X=X_train, ti=ti_train, y=y_train)\r\n\r\n# predict outcomes\r\npreds = model.predict(X_test)\r\n# predict uplift\r\nuplift_preds = preds[:, 1] - preds[:, 0]\r\n\r\n```\r\n\r\n## File Locations\r\n\r\n * `src/*` — C++ code that ultimately compiles into a library\r\n * `include/` — C++ header files\r\n * `python-package/` — python package\r\n \r\n## License\r\n\r\nThis project is open-sourced under the MIT license. You can find the terms of the license [here](http://opensource.org/licenses/MIT).\r\n\r\n## Reference Paper\r\n\r\nJunjie Gao, Xiangyu Zheng, DongDong Wang, Zhixiang Huang, Bangqi Zheng, Kai Yang. \"[UTBoost: A Tree-boosting based System for Uplift Modeling](https://arxiv.org/abs/2312.02573)\".","funding_links":[],"categories":[],"sub_categories":[],"project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fjd-opensource%2Futboost","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fjd-opensource%2Futboost","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fjd-opensource%2Futboost/lists"}