Skip to content

Commit cef94bc

Browse files
authored
Update README.md
1 parent 3df384b commit cef94bc

File tree

1 file changed

+21
-25
lines changed

1 file changed

+21
-25
lines changed

README.md

Lines changed: 21 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)