Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modernize XGBoost Python document. (#7468)

Browse Source 
* Use sphinx gallery to integrate examples.
* Remove mock objects.
* Add dask doc inventory.
This commit is contained in:
Jiaming Yuan
2021-11-23 23:24:52 +08:00
committed by GitHub
parent 96a9848c9e
commit c024c42dce
30 changed files with 130 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
demo/guide-python/custom_rmsle.py
View File

@@ -1,16 +1,19 @@
'''Demo for defining customized metric and objective. Notice that for
simplicity reason weight is not used in following example. In this
script, we implement the Squared Log Error (SLE) objective and RMSLE metric as customized
functions, then compare it with native implementation in XGBoost.
"""
Demo for defining a custom regression objective and metric
==========================================================
See doc/tutorials/custom_metric_obj.rst for a step by step
walkthrough, with other details.
Demo for defining customized metric and objective. Notice that for simplicity reason
weight is not used in following example. In this script, we implement the Squared Log
Error (SLE) objective and RMSLE metric as customized functions, then compare it with
native implementation in XGBoost.
The `SLE` objective reduces impact of outliers in training dataset,
hence here we also compare its performance with standard squared
error.
See doc/tutorials/custom_metric_obj.rst for a step by step walkthrough, with other
details.
'''
The `SLE` objective reduces impact of outliers in training dataset, hence here we also
compare its performance with standard squared error.
"""
import numpy as np
import xgboost as xgb
from typing import Tuple, Dict, List
@@ -171,9 +174,6 @@ def plot_history(rmse_evals, rmsle_evals, py_rmsle_evals):
ax2.plot(x, py_rmsle_evals['dtest']['PyRMSLE'], label='test-PyRMSLE')
ax2.legend()
plt.show()
plt.close()
def main(args):
dtrain, dtest = generate_data()
@@ -183,9 +183,10 @@ def main(args):
if args.plot != 0:
plot_history(rmse_evals, rmsle_evals, py_rmsle_evals)
plt.show()
if __name__ == '__main__':
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description='Arguments for custom RMSLE objective function demo.')
parser.add_argument(