import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
def createData():
df = pd.read_csv('http://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data',
header=None) # 加载Iris数据集作为DataFrame对象
X = df.iloc[:, [0, 2]].values # 取出2个特征,并把它们用Numpy数组表示
X = X[:] # 复制一个切片取出后面100个用例
X = np.c_[X, np.ones(150)] # 矩阵增加一列,常数列
# print(X)
y = np.zeros((150, 2))
for i in range(100):
if i < 50:
y[i][0] = 1
else:
y[i][1] = 1
# print(y)
plt.scatter(X[:50, 0], X[:50, 1], color='red', marker='o', label='setosa') # 前50个样本的散点图
plt.scatter(X[50:100, 0], X[50:100, 1], color='blue', marker='x', label='versicolor') # 中间50个样本的散点图
plt.scatter(X[100:, 0], X[100:, 1], color='green', marker='+'
Softmax,批量梯度下降,随机梯度下降(纯手写不用库)复制就能跑
最新推荐文章于 2022-11-19 17:08:27 发布