ML之catboost:基于自定义数据集利用catboost 算法实现回归预测(训练采用CPU和GPU两种方式)

简介: ML之catboost:基于自定义数据集利用catboost 算法实现回归预测(训练采用CPU和GPU两种方式)


目录

基于自定义数据集利用catboost 算法实现回归预测(训练采用CPU和GPU两种方式)

输出结果

# T1、训练采用CPU

# T2、训练采用GPU

实现代码


基于自定义数据集利用catboost 算法实现回归预测(训练采用CPU和GPU两种方式)

输出结果

# T1、训练采用CPU

1. 0: learn: 7.9608417  total: 50.2ms remaining: 1.46s
2. 1: learn: 7.7618206  total: 50.5ms remaining: 707ms
3. 2: learn: 7.5879985  total: 50.9ms remaining: 458ms
4. 3: learn: 7.4148674  total: 51.4ms remaining: 334ms
5. 4: learn: 7.2488388  total: 51.6ms remaining: 258ms
6. 5: learn: 7.0676178  total: 51.8ms remaining: 207ms
7. 6: learn: 6.8909273  total: 51.9ms remaining: 171ms
8. 7: learn: 6.7186541  total: 52.1ms remaining: 143ms
9. 8: learn: 6.5506878  total: 52.3ms remaining: 122ms
10. 9:  learn: 6.3869206  total: 52.4ms remaining: 105ms
11. 10: learn: 6.2272476  total: 52.6ms remaining: 90.9ms
12. 11: learn: 6.0715664  total: 52.8ms remaining: 79.2ms
13. 12: learn: 5.9197772  total: 52.9ms remaining: 69.2ms
14. 13: learn: 5.7717828  total: 53.1ms remaining: 60.7ms
15. 14: learn: 5.6274882  total: 53.2ms remaining: 53.2ms
16. 15: learn: 5.4868010  total: 53.4ms remaining: 46.7ms
17. 16: learn: 5.3496310  total: 53.6ms remaining: 41ms
18. 17: learn: 5.2158902  total: 53.7ms remaining: 35.8ms
19. 18: learn: 5.0854930  total: 53.9ms remaining: 31.2ms
20. 19: learn: 4.9583556  total: 54ms remaining: 27ms
21. 20: learn: 4.8343967  total: 54.2ms remaining: 23.2ms
22. 21: learn: 4.7135368  total: 54.4ms remaining: 19.8ms
23. 22: learn: 4.5956984  total: 54.5ms remaining: 16.6ms
24. 23: learn: 4.4808059  total: 54.7ms remaining: 13.7ms
25. 24: learn: 4.3687858  total: 54.9ms remaining: 11ms
26. 25: learn: 4.2595661  total: 55ms remaining: 8.46ms
27. 26: learn: 4.1530770  total: 55.2ms remaining: 6.13ms
28. 27: learn: 4.0492501  total: 55.3ms remaining: 3.95ms
29. 28: learn: 3.9480188  total: 55.5ms remaining: 1.91ms
30. 29: learn: 3.8493183  total: 55.7ms remaining: 0us
31. [14.72696421 19.90303684]

# T2、训练采用GPU

1. 0: learn: 7.9608417  total: 114ms  remaining: 3.29s
2. 1: learn: 7.7618210  total: 118ms  remaining: 1.65s
3. 2: learn: 7.5677758  total: 122ms  remaining: 1.1s
4. 3: learn: 7.3785819  total: 125ms  remaining: 814ms
5. 4: learn: 7.1941178  total: 128ms  remaining: 642ms
6. 5: learn: 7.0142648  total: 132ms  remaining: 529ms
7. 6: learn: 6.8389078  total: 136ms  remaining: 446ms
8. 7: learn: 6.6679348  total: 140ms  remaining: 385ms
9. 8: learn: 6.5012368  total: 143ms  remaining: 335ms
10. 9:  learn: 6.3387062  total: 149ms  remaining: 297ms
11. 10: learn: 6.1802387  total: 153ms  remaining: 264ms
12. 11: learn: 6.0257332  total: 156ms  remaining: 234ms
13. 12: learn: 5.8750898  total: 162ms  remaining: 212ms
14. 13: learn: 5.7282121  total: 166ms  remaining: 190ms
15. 14: learn: 5.5850064  total: 171ms  remaining: 171ms
16. 15: learn: 5.4453814  total: 175ms  remaining: 153ms
17. 16: learn: 5.3092462  total: 180ms  remaining: 137ms
18. 17: learn: 5.1765153  total: 184ms  remaining: 122ms
19. 18: learn: 5.0471030  total: 188ms  remaining: 109ms
20. 19: learn: 4.9209255  total: 193ms  remaining: 96.3ms
21. 20: learn: 4.7979018  total: 196ms  remaining: 84.2ms
22. 21: learn: 4.6779541  total: 200ms  remaining: 72.7ms
23. 22: learn: 4.5610056  total: 204ms  remaining: 62.1ms
24. 23: learn: 4.4469804  total: 208ms  remaining: 51.9ms
25. 24: learn: 4.3358058  total: 212ms  remaining: 42.5ms
26. 25: learn: 4.2274105  total: 218ms  remaining: 33.6ms
27. 26: learn: 4.1217256  total: 223ms  remaining: 24.8ms
28. 27: learn: 4.0186824  total: 227ms  remaining: 16.2ms
29. 28: learn: 3.9182151  total: 230ms  remaining: 7.95ms
30. 29: learn: 3.8202597  total: 235ms  remaining: 0us
31. [14.67884178 20.        ]

实现代码

1. # ML之catboost:基于自定义数据集实现回归预测
2. from catboost import CatBoostRegressor
3. 
4. #1、定义数据集
5. train_data = [[12, 14, 16, 18], 
6.               [23, 25, 27, 29], 
7.               [32, 34, 36, 38]]
8. train_labels = [10, 20, 30]
9. 
10. eval_data = [[2, 4, 6, 8], 
11.              [20, 21, 24, 33]]
12. 
13. #2、模型预测
14. model_CatBR = CatBoostRegressor(iterations=30, learning_rate=0.1, depth=2)
15. model_CatBR.fit(train_data, train_labels)
16. preds = model_CatBR.predict(eval_data)
17. print(preds)


相关实践学习
部署Stable Diffusion玩转AI绘画(GPU云服务器)
本实验通过在ECS上从零开始部署Stable Diffusion来进行AI绘画创作,开启AIGC盲盒。
相关文章
|
2月前
|
机器学习/深度学习 算法 数据挖掘
K-means聚类算法是机器学习中常用的一种聚类方法,通过将数据集划分为K个簇来简化数据结构
K-means聚类算法是机器学习中常用的一种聚类方法,通过将数据集划分为K个簇来简化数据结构。本文介绍了K-means算法的基本原理,包括初始化、数据点分配与簇中心更新等步骤,以及如何在Python中实现该算法,最后讨论了其优缺点及应用场景。
124 4
|
19天前
|
存储 算法 数据挖掘
重磅发布 | OpenSearch推出向量检索GPU图算法方案并支持GPU规格售卖
OpenSearch向量检索版推出了面向企业开发者的GPU图算法方案(CAGRA算法),支持客户直接购买GPU规格节点,是国内首家支持GPU规格的向量检索产品。
114 12
|
3月前
|
存储 机器学习/深度学习 算法
蓝桥杯练习题(三):Python组之算法训练提高综合五十题
蓝桥杯Python编程练习题的集合,涵盖了从基础到提高的多个算法题目及其解答。
134 3
蓝桥杯练习题(三):Python组之算法训练提高综合五十题
|
2月前
|
分布式计算 Java 开发工具
阿里云MaxCompute-XGBoost on Spark 极限梯度提升算法的分布式训练与模型持久化oss的实现与代码浅析
本文介绍了XGBoost在MaxCompute+OSS架构下模型持久化遇到的问题及其解决方案。首先简要介绍了XGBoost的特点和应用场景,随后详细描述了客户在将XGBoost on Spark任务从HDFS迁移到OSS时遇到的异常情况。通过分析异常堆栈和源代码,发现使用的`nativeBooster.saveModel`方法不支持OSS路径,而使用`write.overwrite().save`方法则能成功保存模型。最后提供了完整的Scala代码示例、Maven配置和提交命令,帮助用户顺利迁移模型存储路径。
|
3月前
|
机器学习/深度学习 算法 决策智能
【机器学习】揭秘深度学习优化算法:加速训练与提升性能
【机器学习】揭秘深度学习优化算法:加速训练与提升性能
|
3月前
|
算法 Java C++
【贪心算法】算法训练 ALGO-1003 礼物(C/C++)
【贪心算法】算法训练 ALGO-1003 礼物(C/C++)
【贪心算法】算法训练 ALGO-1003 礼物(C/C++)
|
3月前
|
算法 C++
蓝桥 算法训练 共线(C++)
蓝桥 算法训练 共线(C++)
|
5月前
|
数据采集 机器学习/深度学习 算法
【python】python客户信息审计风险决策树算法分类预测(源码+数据集+论文)【独一无二】
【python】python客户信息审计风险决策树算法分类预测(源码+数据集+论文)【独一无二】
|
6月前
|
负载均衡 算法 应用服务中间件
nginx自定义负载均衡及根据cpu运行自定义负载均衡
nginx自定义负载均衡及根据cpu运行自定义负载均衡
114 1
|
5月前
|
算法 搜索推荐
支付宝商业化广告算法问题之基于pretrain—>finetune范式的知识迁移中,finetune阶段全参数训练与部分参数训练的效果如何比较
支付宝商业化广告算法问题之基于pretrain—>finetune范式的知识迁移中,finetune阶段全参数训练与部分参数训练的效果如何比较