机器学习算法导论代码---decision_tree_regressor

import numpy as np
from machine_learning.lib.decision_tree_base import DecisionTreeBase
# 对下标属于集合idx的数据计算标签的方差
def get_var(y,idx):
    y_avg=np.average(y[idx])*np.ones(len(idx))
    return np.linalg.norm(y_avg-y[idx],2)**2/len(idx)
#定义决策树回归算法DecisionTreeRegressor，继承了DecisionTreeBase基类
class DecisionTreeRegressor(DecisionTreeBase):
    def __init__(self,max_depth=0,feature_sasmple_rate=1.0):
        super().__init__(
            max_depth=max_depth,
            feature_sample_rate=feature_sasmple_rate,
            get_score=get_var
        )