使用sklearn构建模型的通用模板和预估对象
1. 通用模板
该文章详细介绍了数据集的加载、训练集测试集的划分、基本算法模型模板、有交叉验证的算法模型模板、可调参的算法模型模板等,有不错的参考价值。例如,下图是该文章的基本算法模型模板:
2. 预估对象
预估对象是指上图中的"3、模型名"。
sklearn中文文档中有一些详细的介绍:
举例:
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_ # 获取聚类准则的总和