From 801647f452e5d0a4de850adb10d48999e30dd300 Mon Sep 17 00:00:00 2001 From: Mike Vink Date: Fri, 25 Sep 2020 09:24:42 +0200 Subject: print change --- tree.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tree.py') diff --git a/tree.py b/tree.py index 6161d33..7ed632e 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=}, bootstraps={len(tr)})' + f'Results from: prediction bagged tree({nmin=}, {minleaf=}, {nfeat=}, trees={len(tr)})' ) else: print( - f'Results from: prediction random forest({nmin=}, {minleaf=}, {nfeat=}, bootstraps={len(tr)})' + f'Results from: prediction random forest({nmin=}, {minleaf=}, {nfeat=}, trees={len(tr)})' ) print( f'\t->Confusion matrix:\n{metrics.confusion_matrix(y, training)}') @@ -397,7 +397,7 @@ if __name__ == '__main__': training=pima_indians[:, 8]) - print('\nDataset: pima indians (takes 2 min max, big data bootstrap)') + print('\nDataset: pima indians') tree_pred_b(x=pima_indians[:, :8], tr=tree_grow_b(x=pima_indians[:, :8], y=pima_indians[:, 8], -- cgit v1.2.3