6.LDA.py 文件源码

def plot_LDA(converted_X,y):
    '''
    plot the graph after transfer
    :param converted_X: train data after transfer
    :param y: train_value
    :return:  None
    '''
    from mpl_toolkits.mplot3d import Axes3D
    fig=plt.figure()
    ax=Axes3D(fig)
    colors='rgb'
    markers='o*s'
    for target,color,marker in zip([0,1,2],colors,markers):
        pos=(y==target).ravel()
        X=converted_X[pos,:]
        ax.scatter(X[:,0], X[:,1], X[:,2],color=color,marker=marker,
            label="Label {0}".format(target))
    ax.legend(loc="best")
    fig.suptitle("Iris After LDA")
    plt.show()