@@ -52,41 +52,39 @@ On the [documentation pages](https://erdogant.github.io/treeplot/) you can find
5252
5353<p align =" left " >
5454 <a href =" https://erdogant.github.io/treeplot/pages/html/Examples.html# " >
55- <img src =" https://github.com/erdogant/treeplot/blob/master/docs/figs/Figure_1 .png " width =" 550 " />
55+ <img src =" https://github.com/erdogant/treeplot/blob/master/docs/figs/fig_breast_randomforest .png " width =" 600 " />
5656 </a >
5757</p >
5858
5959
6060#
6161
62- #### Example XGboost:
63- ``` python
64- # Load example dataset
65- X,y = treeplot.import_example()
66- # Learn model
67- from xgboost import XGBClassifier
68- model = XGBClassifier(n_estimators = 100 , max_depth = 2 , random_state = 0 ).fit(X, y)
69- ```
7062
71- ``` python
72- # Make plot
73- ax = treeplot.plot(model)
74- # or directly
75- ax = treeplot.xgboost(model)
76- ```
77- <p align =" center " >
78- <img src =" https://github.com/erdogant/treeplot/blob/master/docs/figs/Figure2_xgboost_hor.png " width =" 550 " />
79- <img src =" https://github.com/erdogant/treeplot/blob/master/docs/figs/Figure2_featimportance.png " width =" 350 " />
63+ * [ Example: XGboot] ( https://erdogant.github.io/treeplot/pages/html/Examples.html#xgboost )
64+
65+ <p align =" left " >
66+ <a href =" https://erdogant.github.io/treeplot/pages/html/Examples.html#xgboost " >
67+ <img src =" https://github.com/erdogant/treeplot/blob/master/docs/figs/fig_breast_xgboot_tree.png " width =" 600 " />
68+ <br >
69+ <img src =" https://github.com/erdogant/treeplot/blob/master/docs/figs/fig_breast_xgboot_weights.png " width =" 600 " />
70+ </a >
8071</p >
8172
82- ``` python
83- # If more parameters needs to be specified, use the exact function:
84- ax = treeplot.xgboost(model, plottype = ' vertical' )
85- ```
86- <p align =" center " >
73+
74+ <p align =" left " >
75+ <a href =" https://erdogant.github.io/treeplot/pages/html/Examples.html# " >
8776 <img src =" https://github.com/erdogant/treeplot/blob/master/docs/figs/Figure2_xgboost_ver.png " width =" 550 " />
77+ </a >
8878</p >
8979
80+ #
81+ * [ Example: gradientboostingclassifier] ( https://erdogant.github.io/treeplot/pages/html/Examples.html#gradientboostingclassifier )
82+ #
83+ * [ Example: lightbm] ( https://erdogant.github.io/treeplot/pages/html/Examples.html#lightbm )
84+ #
85+ * [ Example: Explore other trees such as second best tree etc] ( https://erdogant.github.io/treeplot/pages/html/Examples.html#plot-second-best-tree-and-other-trees )
86+
87+ <hr >
9088
9189#### Maintainers
9290* Erdogan Taskesen, github: [ erdogant] ( https://github.com/erdogant )
@@ -95,5 +93,3 @@ ax = treeplot.xgboost(model, plottype='vertical')
9593* Contributions are welcome.
9694* If you wish to buy me a <a href =" https://www.buymeacoffee.com/erdogant " >Coffee</a > for this work, it is very appreciated :)
9795
98- #### Licence
99- See [ LICENSE] ( LICENSE ) for details.
0 commit comments