From 13b9dfaad26758fca3b673c5e64ee266db82568c Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Fri, 8 Dec 2023 14:45:59 +0100 Subject: README: Document OMP_NUM_THREADS for xgboost --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 12ab0ed..f2ddb75 100644 --- a/README.md +++ b/README.md @@ -113,6 +113,7 @@ The following variables may be set to alter the behaviour of dfatool components. | `DFATOOL_USE_XGBOOST` | **0**, 1 | Use Extreme Gradient Boosting algorithm for decision forest generation. | | `DFATOOL_XGB_N_ESTIMATORS` | 1 .. **100** .. *n* | Number of estimators (i.e., trees) for XGBoost. | | `DFATOOL_XGB_MAX_DEPTH` | 2 .. **10** ** *n* | Maximum XGBoost tree depth. | +| `OMP_NUM_THREADS` | *number of CPU cores* | Maximum number of threads used per XGBoost learner. A limit of 4 threads appears to be ideal. Note that dfatool may spawn several XGBoost instances at the same time. | | `DFATOOL_KCONF_IGNORE_NUMERIC` | **0**, 1 | Ignore numeric (int/hex) configuration options. Useful for comparison with CART/DECART. | | `DFATOOL_KCONF_IGNORE_STRING` | **0**, 1 | Ignore string configuration options. Useful for comparison with CART/DECART. | | `DFATOOL_FIT_LINEAR_ONLY` | **0**, 1 | Only consider linear functions (a + bx) in regression analysis. Useful for comparison with Linear Model Trees / M5. | -- cgit v1.2.3