37,719
社区成员
发帖
与我相关
我的任务
分享
#报错信息
Traceback (most recent call last):
File "c:\Users\Asura-Dong\Desktop\Blog\learn\MachineLearning\Unit5\Logistic.py", line 99, in <module>
weights1 = stocGradAscent1(dataArr,labelMat)
File "c:\Users\Asura-Dong\Desktop\Blog\learn\MachineLearning\Unit5\Logistic.py", line 78, in stocGradAscent1
weights = weights+alpha*error*dataMatrix[randIndex]
TypeError: 'numpy.float64' object cannot be interpreted as an integer
from numpy import *
from matplotlib import pyplot as plt
def loadDataSet():
dataMat = []
labelMat = []
fr = open('testSet.txt')
for line in fr.readlines():
lineArr = line.strip().split()
dataMat.append([1.0, float(lineArr[0]), float(lineArr[1])])
labelMat.append(int(lineArr[2]))
return dataMat,labelMat
#定义Sigmoid函数
def sigmoid(inX):
return 1.0/(1+exp(-inX))
def gradAscent(dataMatIn,classLabels):
dataMatrix = mat(dataMatIn)
labelMat = mat(classLabels).transpose()
m,n = shape(dataMatrix) #m个样本,n维/个
alpha = 0.001#移动的步长
maxCycles = 500 #迭代次数
weights = ones((n,1))
for k in range(maxCycles):
h = sigmoid(dataMatrix*weights)
error = (labelMat-h)
weights = weights + alpha*dataMatrix.T*error
return weights
def plotBestFit(weights):
dataMat,labelMat = loadDataSet()
dataArr = array(dataMat)
n = shape(dataArr)[0]
xcord1 = []
ycord1 = []
xcord2 = []
ycord2 = []
for i in range(n):
if int(labelMat[i])==1:
xcord1.append(dataArr[i,1])
ycord1.append(dataArr[i,2])
else:
xcord2.append(dataArr[i,1])
ycord2.append(dataArr[i,2])
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(xcord1,ycord1,s=30,c='red',marker='s')
ax.scatter(xcord2,ycord2,s=30,c='green')
x = arange(-3.0,3.0,0.1)
y = (-weights[0]-weights[1]*x)/weights[2]
ax.plot(x,y)
plt.show()
def stocGradAscent0(dataMatrix,classLabels):
m,n = shape(dataMatrix)
alpha = 0.01
weights = ones(n)
for i in range(m):
h = sigmoid(dataMatrix[i]*weights)
error = classLabels[i]-h
#error和h是数值
weights = weights + alpha*error*dataMatrix[i]
return weights
def stocGradAscent1(dataMatrix,classLabels,numIter = 150):
m,n = shape(dataMatrix)
weights = array([1.0,1.0,1.0])
for j in range(numIter):
dataIndex = range(m)
for i in range(m):
alpha = 4//(1.0+j+i)+0.01 #注意常数项.保证alpha不断变小,但不为0
randIndex = int(random.uniform(0,len(dataIndex)))#改动2:随机选取
h = sigmoid(sum(dataMatrix[randIndex]*weights))
error = classLabels[randIndex]-h
weights = weights+alpha*error*dataMatrix[randIndex]
del(dataIndex[randIndex])
return weights
if __name__=='__main__':
dataArr , labelMat = loadDataSet()
'''
print("dataArr is:")
for line in dataArr:
print(line)
print("labelMat is:\n",labelMat)
for line in labelMat:
print(line)
weights = stocGradAscent0(dataArr,labelMat)
print(weights)
plotBestFit(weights)
'''
weights1 = stocGradAscent1(dataArr,labelMat)
print(weights)
plotBestFit(weights)