最近斯坦福大学,又搞出了一个新的树。
首先安装,斯坦福的ngboost框架
pip install ngboost然后给出一个案例代码。X和Y对应的是两个numpy矩阵可以替换成自己的数据集。
from ngboost.ngboost import NGBoost
from ngboost.learners import default_tree_learner
from ngboost.scores import MLE
from ngboost.distns import Normal
from sklearn.datasets import load_boston
from sklearn.model_selection import trAIn_test_split
from sklearn.metrics import mean_squared_error
# 获取波士顿数据集
X, Y = load_boston(True)
# 随机切分数据集
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.2)