NGBoost:用于概率预测的自然梯度提升
ngboost是一个Python库,实现了“自然梯度增强”,如。 它建立在,旨在针对适当的评分规则,分布和基础学习者的选择进行扩展和模块化。 在此可以对NGBoost的基本方法进行详尽的介绍。
安装
via pip
pip install --upgrade ngboost
via conda-forge
conda install -c conda-forge ngboost
用法
波士顿住房数据集上的概率回归示例:
from ngboost import NGBRegressor
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
1