決策樹(二)決策樹迴歸

迴歸 決策樹也可以用於執行迴歸任務。我們首先用sk-learn的DecisionTreeRegressor類構造一顆迴歸決策樹,並在一個帶噪聲的二次方數據集上進行訓練,指定max_depth=2: import numpy as np # Quadratic training set + noise np.random.seed(42) m = 200 X = np.random.rand(m
相關文章
相關標籤/搜索