sklearn神经网络分类

时间：2022-12-02 15:36:10

sklearn神经网络分类

神经网络学习能力强大，在数据量足够，隐藏层足够多的情况下，理论上可以拟合出任何方程。

理论部分

sklearn提供的神经网络算法有三个:

neural_network.BernoulliRBM，neural_network.MLPClassifier，neural_network.MLPRgression

我们现在使用MLP(Multi-Layer Perception)做分类，回归其实也类似。该网络由三部分组成：输入层、隐藏层、输出层，其中隐藏层的个数可以人为设定。神经网络学习之后的知识都存在每一层的权重矩阵中，学习的过程也就是不断训练权重达到拟合的效果。权重训练比较常用的方法是反向传递(Backpropagation)

sklearn神经网络分类

分类代码

#coding=utf-8

import pandas as pd

import matplotlib.pyplot as plt

from sklearn.model_selection import train_test_split

from sklearn import datasets

from sklearn.neural_network import MLPClassifier

import numpy as np

from sklearn.preprocessing import StandardScaler

def main():

    iris = datasets.load_iris() #典型分类数据模型

    #这里我们数据统一用pandas处理

    data = pd.DataFrame(iris.data, columns=iris.feature_names)

    data['class'] = iris.target

    #这里只取两类

    data = data[data['class']!=2]

    #为了可视化方便，这里取两个属性为例

    scaler = StandardScaler()

    X = data[['sepal length (cm)','sepal width (cm)']]

    scaler.fit(X)

    #标准化数据集

    X = scaler.transform(X)

    Y = data[['class']]

    #划分数据集

    X_train, X_test, Y_train, Y_test =train_test_split(X, Y)

    mpl = MLPClassifier(solver='lbfgs',activation='logistic')

    mpl.fit(X_train, Y_train)

    print 'Score:\n',mpl.score(X_test, Y_test) #score是指分类的正确率

    #区域划分

    h = 0.02

    x_min, x_max = X[:, 0].min() - 1, X[:, 0].max() + 1

    y_min, y_max = X[:, 1].min() - 1, X[:, 1].max() + 1

    xx, yy = np.meshgrid(np.arange(x_min, x_max, h),

                         np.arange(y_min, y_max, h))

    Z = mpl.predict(np.c_[xx.ravel(), yy.ravel()])

    Z = Z.reshape(xx.shape)

    plt.contourf(xx, yy, Z, cmap=plt.cm.Paired)

    #做出原来的散点图

    class1_x = X[Y['class']==0,0]

    class1_y = X[Y['class']==0,1]

    l1 = plt.scatter(class1_x,class1_y,color='b',label=iris.target_names[0])

    class2_x = X[Y['class']==1,0]

    class2_y = X[Y['class']==1,1]

    l2 = plt.scatter(class2_x,class2_y,color='r',label=iris.target_names[1])

    class3_x = X[Y['class']==2,0]

    class3_y = X[Y['class']==2,1]

    l3 = plt.scatter(class3_x,class3_y,color='g',label=iris.target_names[2])

    plt.legend(handles = [l1, l2,l3], loc = 'best')

    plt.grid(True)

    plt.show()

if __name__ == '__main__':

    main()

测试结果

sklearn神经网络分类

相关文章