From 919a14df6d3afc4657fcd6359319f08427f3a90c Mon Sep 17 00:00:00 2001 From: Mike Vink Date: Fri, 25 Sep 2020 09:20:10 +0200 Subject: Minor: print bootstraps used for prediction --- tree.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tree.py b/tree.py index 75ea633..b3127ce 100644 --- a/tree.py +++ b/tree.py @@ -345,11 +345,11 @@ def tree_pred_b(x=None, tr=None, training=None, **defaults) -> np.array: # print(np.mean(training == y)) if nfeat == x.shape[1]: print( - f'Results from: prediction bagged tree({nmin=}, {minleaf=}, {nfeat=})' + f'Results from: prediction bagged tree({nmin=}, {minleaf=}, {nfeat=}, bootstraps={len(tr)})' ) else: print( - f'Results from: prediction random forest tree({nmin=}, {minleaf=}, {nfeat=})' + f'Results from: prediction random forest({nmin=}, {minleaf=}, {nfeat=}, bootstraps={len(tr)})' ) print( f'\t->Confusion matrix:\n{metrics.confusion_matrix(y, training)}') -- cgit v1.2.3