summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMike Vink <mike1994vink@gmail.com>2020-10-09 22:26:35 +0200
committerMike Vink <mike1994vink@gmail.com>2020-10-09 22:26:35 +0200
commit817757e750199df1c3f782c823ed1742c0403f98 (patch)
tree33fdd60550f1c03b88e8cacbcb819c703d709c74
parent5eeb3766fa3b94a58c23485664b18e0cfa92023b (diff)
Commit
-rw-r--r--tree.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/tree.py b/tree.py
index 0351e3c..db6975b 100644
--- a/tree.py
+++ b/tree.py
@@ -203,10 +203,13 @@
# 2D data array corresponding to some node
# /y/ numpy.array, 1D numpy array of binary labels corresponding to the
# rows in the 2D data array corresponding to some node
-# /minleaf/ @todo
+# /minleaf/ int, number of x-rows a child must have before splitting
# Returns -> tuple
+# Computes the best split based on the given features using the impurity
+# function.
+
# """