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

Fix mistakes in function comments

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-11-11 19:14:36 +01:00
parent 1c869e154e
commit 475ad7e752
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
stree/Strees.py
View File

@@ -284,9 +284,8 @@ class Splitter:
:type data: np.array (m, n_classes) :type data: np.array (m, n_classes)
:param y: vector of labels (classes) :param y: vector of labels (classes)
:type y: np.array (m,) :type y: np.array (m,)
:return: vector with the class assigned to each sample values :return: column of dataset to be taken into account to split dataset
(can be 0, 1, ...) -1 if none produces information gain :rtype: int
:rtype: np.array shape (m,)
""" """
max_gain = 0 max_gain = 0
selected = -1 selected = -1
@@ -307,8 +306,8 @@ class Splitter:
:type data: np.array (m, n_classes) :type data: np.array (m, n_classes)
:param y: vector of labels (classes) :param y: vector of labels (classes)
:type y: np.array (m,) :type y: np.array (m,)
:return: vector with distances to hyperplane (can be positive or neg.) :return: column of dataset to be taken into account to split dataset
:rtype: np.array shape (m,) :rtype: int
""" """
# select the class with max number of samples # select the class with max number of samples
_, samples = np.unique(y, return_counts=True) _, samples = np.unique(y, return_counts=True)