设为首页|收藏本站|
开启左侧

[问答] ngboost 实验波士顿房价预测

[复制链接]
92723 3
噬心__焦焦 发表于 2021-11-19 22:40:40 | 只看该作者 打印 上一主题 下一主题
 
最近斯坦福大学,又搞出了一个新的树。
首先安装,斯坦福的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)

ngb = NGBoost(Base=default_tree_learner, Dist=Normal, Score=MLE(), natural_gradient=True,
              verbose=False)
# 拟合
ngb.fit(X_train, Y_train)
# 预测
Y_preds = ngb.predict(X_test)
Y_dists = ngb.pred_dist(X_test)

# 检验均方误差 test Mean Squared Error
test_MSE = mean_squared_error(Y_preds, Y_test)
print('Test MSE', test_MSE)

# 检验负对数似然test Negative Log Likelihood
test_NLL = -Y_dists.logpdf(Y_test.flatten()).mean()
print('Test NLL', test_NLL)


上一篇:WLSA上海学校十月招生说明会
下一篇:破解核糖起源之谜:南大学者首次提出普适性史前化学模型 ...
@



1.西兔生活网 CTLIVES 内容全部来自网络;
2.版权归原网站或原作者所有;
3.内容与本站立场无关;
4.若涉及侵权或有疑义,请点击“举报”按钮,其他联系方式或无法及时处理。
 

精彩评论3

正序浏览
跳转到指定楼层
沙发
凰十三娘 发表于 2021-11-19 22:41:39 | 只看该作者
 
这个效果怎么样?
回复 支持 反对

使用道具 举报

 
板凳
琳琳女孩仓r5 发表于 2021-11-19 22:41:48 | 只看该作者
 
上线效果不稳定
回复 支持 反对

使用道具 举报

 
地板
打酱油的饿nice 发表于 2021-11-19 22:41:55 | 只看该作者
 
报错拉
回复 支持 反对

使用道具 举报

 
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

排行榜
活跃网友
返回顶部快速回复上一主题下一主题返回列表APP下载手机访问
Copyright © 2016-2028 CTLIVES.COM All Rights Reserved.  西兔生活网  小黑屋| GMT+8, 2024-5-3 15:25