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)


相关实践学习
基于阿里云DeepGPU实例,用AI画唯美国风少女
本实验基于阿里云DeepGPU实例,使用aiacctorch加速stable-diffusion-webui,用AI画唯美国风少女,可提升性能至高至原性能的2.6倍。
相关文章
|
6天前
|
机器学习/深度学习 存储 PyTorch
【AMP实操】解放你的GPU运行内存!在pytorch中使用自动混合精度训练
【AMP实操】解放你的GPU运行内存!在pytorch中使用自动混合精度训练
78 0
|
6天前
|
机器学习/深度学习 弹性计算 TensorFlow
阿里云GPU加速:大模型训练与推理的全流程指南
随着深度学习和大规模模型的普及,GPU成为训练和推理的关键加速器。本文将详细介绍如何利用阿里云GPU产品完成大模型的训练与推理。我们将使用Elastic GPU、阿里云深度学习镜像、ECS(云服务器)等阿里云产品,通过代码示例和详细说明,带你一步步完成整个流程。
1007 0
|
6天前
|
机器学习/深度学习 异构计算 Python
Bert-vits2最终版Bert-vits2-2.3云端训练和推理(Colab免费GPU算力平台)
对于深度学习初学者来说,JupyterNoteBook的脚本运行形式显然更加友好,依托Python语言的跨平台特性,JupyterNoteBook既可以在本地线下环境运行,也可以在线上服务器上运行。GoogleColab作为免费GPU算力平台的执牛耳者,更是让JupyterNoteBook的脚本运行形式如虎添翼。 本次我们利用Bert-vits2的最终版Bert-vits2-v2.3和JupyterNoteBook的脚本来复刻生化危机6的人气角色艾达王(ada wong)。
Bert-vits2最终版Bert-vits2-2.3云端训练和推理(Colab免费GPU算力平台)
|
6天前
|
机器学习/深度学习 人工智能 算法
为什么大模型训练需要GPU,以及适合训练大模型的GPU介绍
为什么大模型训练需要GPU,以及适合训练大模型的GPU介绍
23 0
|
6天前
|
机器学习/深度学习 并行计算 PyTorch
【多GPU炼丹-绝对有用】PyTorch多GPU并行训练:深度解析与实战代码指南
本文介绍了PyTorch中利用多GPU进行深度学习的三种策略:数据并行、模型并行和两者结合。通过`DataParallel`实现数据拆分、模型不拆分,将数据批次在不同GPU上处理;数据不拆分、模型拆分则将模型组件分配到不同GPU,适用于复杂模型;数据和模型都拆分,适合大型模型,使用`DistributedDataParallel`结合`torch.distributed`进行分布式训练。代码示例展示了如何在实践中应用这些策略。
161 2
【多GPU炼丹-绝对有用】PyTorch多GPU并行训练:深度解析与实战代码指南
|
6天前
|
机器学习/深度学习 缓存 PyTorch
Yolov5如何训练自定义的数据集,以及使用GPU训练,涵盖报错解决
Yolov5如何训练自定义的数据集,以及使用GPU训练,涵盖报错解决
488 0
|
6天前
|
机器学习/深度学习 人工智能 弹性计算
阿里云林立翔:基于阿里云GPU的AIGC小规模训练优化方案
阿里云弹性计算林立翔在【AIGC】话题下带来了题为《基于阿里云GPU的AIGC小规模训练优化方案》的主题演讲,围绕生成式AI技术栈、生成式AI微调训练和性能分析、ECS GPU实例为生成式AI提供算力保障、应用场景案例等相关话题展开。
|
5月前
|
XML 数据格式 异构计算
笔记 ubuntu18.04安装cuda10.2 cudnn7.5,然后进行物体检测gpu训练
笔记 ubuntu18.04安装cuda10.2 cudnn7.5,然后进行物体检测gpu训练
53 1
|
6天前
|
人工智能 机器人 Serverless
魔搭大模型一键部署到阿里云函数计算,GPU 闲置计费功能可大幅降低开销
魔搭大模型一键部署到阿里云函数计算,GPU 闲置计费功能可大幅降低开销
604 2
|
6天前
|
负载均衡 固态存储 Linux
阿里云轻量应用服务器、云服务器、gpu云服务器最新收费标准参考
轻量应用服务器、云服务器、gpu云服务器是阿里云服务器产品中,比较热门的云服务器产品类型,不同类型的云服务器产品收费模式与收费标准是不一样的,本文为大家展示这几个云服务器产品的最新收费标准情况,以供参考。
阿里云轻量应用服务器、云服务器、gpu云服务器最新收费标准参考