INNER CODE UNIT · Python
__init__
ankonzoid/LearningX · basic_ML/classification/decision_tree.py:12
def __init__(self):
pass
def fit(self, X, y):
def loss_func(X, y):
# Gini impurity
p2 = 0.0
y_classes = list(set(y))
for c in y_classes:
p2 += (np.sum(y == c) / len(y)) ** 2
loss = 1.0 - p2
return loss
def perform_split(X, y):
N, d = X.shape
loss_best = 9E99
X_left, y_left = [], []