الأشجار وتقسيم الفضاء

import numpy as np
x = np.array([1.0, 2.0, 3.0, 4.0])
y = np.array([0, 0, 1, 1])
def gini(values):
    p = np.mean(values == 1)
    return 1.0 - p * p - (1.0 - p) ** 2
def split_impurity(threshold):
    left = x <= threshold
    assert left.any() and (~left).any()
    return left.mean() * gini(y[left]) + (~left).mean() * gini(y[~left])
for threshold in [1.5, 2.5]:
    print(f"threshold={threshold:.1f} impurity={split_impurity(threshold):.3f}")

threshold=1.5 impurity=0.333
threshold=2.5 impurity=0.000

يكون الموزون ثلثا للأول وصفرا للفصل المصطنع التام.

import numpy as np
future = np.array([1.5, 2.5, 3.5])
labels = np.where(future <= 2.5, 0, 1)
print(labels.tolist())

[0, 0, 1]

يتبع التعادل اليسار؛ يطبق المثال جذعا ثابتا لا مصنفا مدربا جديدا أو ادعاء جودة.