Skip to content

Commit 5cc5528

Browse files
authored
implemented xgdboost_regression (#314)
* implemented xgd_regression
1 parent d459c48 commit 5cc5528

3 files changed

Lines changed: 779 additions & 0 deletions

File tree

src/lib.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,5 +130,6 @@ pub mod readers;
130130
pub mod svm;
131131
/// Supervised tree-based learning methods
132132
pub mod tree;
133+
pub mod xgboost;
133134

134135
pub(crate) mod rand_custom;

src/xgboost/mod.rs

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
//! # XGBoost
2+
//!
3+
//! XGBoost, which stands for Extreme Gradient Boosting, is a powerful and efficient implementation of the gradient boosting framework. Gradient boosting is a machine learning technique for regression and classification problems, which produces a prediction model in the form of an ensemble of weak prediction models, typically decision trees.
4+
//!
5+
//! The core idea of boosting is to build the model in a stage-wise fashion. It learns from its mistakes by sequentially adding new models that correct the errors of the previous ones. Unlike bagging, which trains models in parallel, boosting is a sequential process. Each new tree is fit on a modified version of the original data set, specifically focusing on the instances where the previous models performed poorly.
6+
//!
7+
//! XGBoost enhances this process through several key innovations. It employs a more regularized model formalization to control over-fitting, which gives it better performance. It also has a highly optimized and parallelized tree construction process, making it significantly faster and more scalable than traditional gradient boosting implementations.
8+
//!
9+
//! ## References:
10+
//!
11+
//! * "Elements of Statistical Learning", Hastie T., Tibshirani R., Friedman J., 10. Boosting and Additive Trees
12+
//! * XGBoost: A Scalable Tree Boosting System, Chen T., Guestrin C.
13+
14+
// xgboost implementation
15+
pub mod xgb_regressor;
16+
pub use xgb_regressor::{XGRegressor, XGRegressorParameters};

0 commit comments

Comments
 (0)