Skip to content

Commit 52b64eb

Browse files
committed
Default number of trees in plot
1 parent 2e7ff47 commit 52b64eb

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

nestedhyperboost/results.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,8 @@ def feat_plot(self):
3333
## feature importance plot
3434
def tree_plot(self):
3535
plot_tree(
36-
booster = self.model,
37-
num_trees = 0
38-
)
36+
booster = self.model
37+
)
3938

4039
## store regression results
4140
class RegressResults(Results):

0 commit comments

Comments
 (0)