Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-16 07:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

#2 update benchmark notebook

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-06-15 10:33:51 +02:00
parent c94bc068bd
commit 736ab7ef20
2 changed files with 48 additions and 210 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
stree/Strees.py
View File

@@ -193,8 +193,8 @@ class Splitter:
def information_gain(
self, labels_up: np.array, labels_dn: np.array
) -> float:
card_up = labels_up.shape[0]
card_dn = labels_dn.shape[0]
card_up = labels_up.shape[0] if labels_up is not None else 0
card_dn = labels_dn.shape[0] if labels_dn is not None else 0
samples = card_up + card_dn
up = card_up / samples * self.criterion_function(labels_up)
dn = card_dn / samples * self.criterion_function(labels_dn)