使用sklearn构建模型的通用模板和预估对象

1. 通用模板

参考:使用sklearn构建模型的通用模板

该文章详细介绍了数据集的加载、训练集测试集的划分、基本算法模型模板、有交叉验证的算法模型模板、可调参的算法模型模板等,有不错的参考价值。例如,下图是该文章的基本算法模型模板:

2. 预估对象

预估对象是指上图中的"3、模型名"。

sklearn中文文档中有一些详细的介绍:

举例:

1
2
3
4
5
6
from sklearn.cluster import KMeans
km = KMeans(n_clusters=6)
km.fit(X)    # X是训练数据
centroids = km.cluster_centers_  # 获取聚类中心
label_pred = km.labels_       # 获取聚类标签
inertia = km.inertia_         # 获取聚类准则的总和

 

posted @   Picassooo  阅读(351)  评论(0编辑  收藏  举报
编辑推荐:
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
阅读排行:
· Manus爆火,是硬核还是营销?
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
点击右上角即可分享
微信分享提示