一.鸢尾花数据集算法可视化
1.LDA对鸢尾花数据集聚类
代码如下:
import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
def LDA(X, y):
#根据y等于0或1分类
X1 = np.array([X[i] for i in range(len(X)) if y[i] == 0])
X2 = np.array([X[i] for i in range(len(X)) if y[i] == 1])
len1 = len(X1)
len2 = len(X2)
mju1 = np.mean(X1, axis=0)#求中心点
mju2 = np.mean(X2, axis=0)
cov1 = np.dot((X1 - mju1).T, (X1 - mju1))
cov2=np.dot((X2 - mju2).T, (X2 - mju2))
Sw = cov1 + cov2
a=mju1-mju2
a=(np.array([a])).T
#计算w
w=(np.dot(np.linalg.inv(Sw),a))
#计算投影直线
#k=w[1]/w[0]
#b=0;
#x=np.arange(0,5)
#yy=k*x+b
#plt.plot(x,yy)
X1_new =func(X1, w)
X2_new = func(X2, w)
y1_new = [1 for i in range(len1)]
y2_new = [2 for i in range(len2)]
return X1_new,X2_new,y1_new,y2_new
def func(x, w):
return np.dot((x), w)
iris = datasets.load_iris()
X = iris["data"][:, (2, 3)] # 花瓣长度与花瓣宽度 petal length, petal width
y = iris["target"]
#print(y)
setosa_or_versicolor = (y == 0) | (y == 1)
X = X[setosa_or_versicolor]
y = y[setosa_or_versicolor]
#print(Sw)
x1_new, X2_new, y1_new, y2_new = LDA(X