zoukankan      html  css  js  c++  java
  • 使用sklearn估计器构建K-Means聚类模型

    实例要求:以sklearn库自带的iris数据集为例,使用sklearn估计器构建K-Means聚类模型,并且完成预测类别功能以及聚类结果可视化。

    实例代码:

    import pandas as pd
    import matplotlib.pyplot as plt
    from sklearn.datasets import load_iris
    from sklearn.preprocessing import MinMaxScaler
    from sklearn.cluster import KMeans
    from sklearn.manifold import TSNE
    
    '''  构建K-Means模型  '''
    iris = load_iris()
    iris_data = iris['data'] # 提取数据集中的数据
    iris_target = iris['target'] # 提取数据集中的标签
    iris_names = iris['feature_names'] # 提取特征名
    scale = MinMaxScaler().fit(iris_data) # 训练规则
    iris_dataScale = scale.transform(iris_data) # 应用规则
    kmeans = KMeans(n_clusters=3,random_state=123).fit(iris_dataScale) # 构建并训练模型
    print('构建的K-Means模型为:
    ',kmeans)
    
    result = kmeans.predict([[1.5,1.5,1.5,1.5]])
    print('花瓣花萼长度宽度全为1.5的鸢尾花预测类别为:',result[0])
    
    '''  聚类结果可视化  '''
    tsne = TSNE(n_components=2,init='random',random_state=177).fit(iris_data)    # 使用TSNE进行数据降维,降成两维
    df = pd.DataFrame(tsne.embedding_)                    # 将原始数据转换为DataFrame
    df['labels'] = kmeans.labels_     # 将聚类结果存储进df数据表中
    df1 = df[df['labels']==0]
    df2 = df[df['labels']==1]
    df3 = df[df['labels']==2]
    # fig = plt.figure(figsize=(9,6))    # 绘制图形  设定空白画布,并制定大小
    plt.plot(df1[0],df1[1],'bo',df2[0],df2[1],'r*',df3[0],df3[1],'gD')
    plt.show()                          # 显示图片

    实例结果:

      构建的K-Means模型为:

      

      花瓣预测结果:

      

      聚类结果可视化:

      

  • 相关阅读:
    SQL 游标示例
    在与SQL Server 建立 连接时出现与网络相关的或特定于实例的错误。未找到或无法访问服务器
    Jquery 设置焦点
    MVC Hidden用法
    Jquery根据name取得所有选中的Checkbox值
    MVC程序部署后页面指向login.aspx
    ref 和out 关键字
    认识关系型数据库
    抽象类和接口详解
    1.穷举法
  • 原文地址:https://www.cnblogs.com/xiaoyh/p/10187195.html
Copyright © 2011-2022 走看看