|
1 | | -############ |
2 | | -DART booster |
3 | | -############ |
| 1 | +#### |
| 2 | +DART |
| 3 | +#### |
4 | 4 | XGBoost mostly combines a huge number of regression trees with a small learning rate. |
5 | 5 | In this situation, trees added early are significant and trees added late are unimportant. |
6 | 6 |
|
7 | 7 | Vinayak and Gilad-Bachrach proposed a new method to add dropout techniques from the deep neural net community to boosted trees, and reported better results in some situations. |
8 | 8 |
|
9 | | -This is a instruction of new tree booster ``dart``. |
| 9 | +This is a instruction of the dropout mode for tree models. Dropout is controlled by |
| 10 | +parameters like ``rate_drop``. The legacy ``dart`` booster name remains available for |
| 11 | +compatibility. |
10 | 12 |
|
11 | 13 | ************** |
12 | 14 | Original paper |
@@ -48,7 +50,8 @@ How it works |
48 | 50 | Parameters |
49 | 51 | ********** |
50 | 52 |
|
51 | | -The booster ``dart`` inherits ``gbtree`` booster, so it supports all parameters that ``gbtree`` does, such as ``eta``, ``gamma``, ``max_depth`` etc. |
| 53 | +Dropout uses the same tree parameters as ``gbtree``, such as ``eta``, ``gamma``, |
| 54 | +``max_depth``, and others. |
52 | 55 |
|
53 | 56 | Additional parameters are noted below: |
54 | 57 |
|
@@ -99,8 +102,7 @@ Sample Script |
99 | 102 | dtrain = xgb.DMatrix('demo/data/agaricus.txt.train?format=libsvm') |
100 | 103 | dtest = xgb.DMatrix('demo/data/agaricus.txt.test?format=libsvm') |
101 | 104 | # specify parameters via map |
102 | | - param = {'booster': 'dart', |
103 | | - 'max_depth': 5, 'learning_rate': 0.1, |
| 105 | + param = {'max_depth': 5, 'learning_rate': 0.1, |
104 | 106 | 'objective': 'binary:logistic', |
105 | 107 | 'sample_type': 'uniform', |
106 | 108 | 'normalize_type': 'tree', |
|
0 commit comments