This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new f692fc7 Improve XGBTuner document (#8428)
f692fc7 is described below
commit f692fc7b730c48747e7d64242e75cd892c7bb47f
Author: ERROR <[email protected]>
AuthorDate: Sat Jul 10 00:50:07 2021 +0700
Improve XGBTuner document (#8428)
* chore: improve xgboost_tuner docstring
* chore: remove whitespace
Co-authored-by: Siwa <[email protected]>
---
python/tvm/autotvm/tuner/xgboost_tuner.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/python/tvm/autotvm/tuner/xgboost_tuner.py
b/python/tvm/autotvm/tuner/xgboost_tuner.py
index 2f4d0ee..9dec54c 100644
--- a/python/tvm/autotvm/tuner/xgboost_tuner.py
+++ b/python/tvm/autotvm/tuner/xgboost_tuner.py
@@ -55,7 +55,9 @@ class XGBTuner(ModelBasedTuner):
The cost model predicts relative rank score.
num_threads: int, optional
- The number of threads. optimizer: str or ModelOptimizer, optional
+ The number of threads.
+
+ optimizer: str or ModelOptimizer, optional
If is 'sa', use a default simulated annealing optimizer.
Otherwise it should be a ModelOptimizer object.
