CS_ALearn/ailearning
1
0
Fork 0
mirror of https://github.com/apachecn/ailearning.git synced 2026-02-13 23:35:28 +08:00
Code Issues Projects Releases Wiki Activity

更新 AdaBoost 的项目内容

Browse Source
This commit is contained in:
jiangzhonglian
2017-08-18 16:30:10 +08:00
parent bd9612ab5b
commit 2b01058049
2 changed files with 156 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/python/7.AdaBoost/adaboost.py
View File

@@ -209,8 +209,8 @@ def plotROC(predStrengths, classLabels):
predStrengths 最终预测结果的权重值
classLabels 原始数据的分类结果集
"""
print 'predStrengths=', predStrengths
print 'classLabels=', classLabels
# print 'predStrengths=', predStrengths
# print 'classLabels=', classLabels
import matplotlib.pyplot as plt
# variable to calculate AUC
@@ -225,7 +225,7 @@ def plotROC(predStrengths, classLabels):
# get sorted index, it's reverse
sortedIndicies = predStrengths.argsort()
# 测试结果是否是从小到大排列
print 'sortedIndicies=', sortedIndicies, predStrengths[0, 176], predStrengths.min(), predStrengths[0, 293], predStrengths.max()
# print 'sortedIndicies=', sortedIndicies, predStrengths[0, 176], predStrengths.min(), predStrengths[0, 293], predStrengths.max()
# 开始创建模版对象
fig = plt.figure()
@@ -244,7 +244,7 @@ def plotROC(predStrengths, classLabels):
ySum += cur[1]
# draw line from cur to (cur[0]-delX, cur[1]-delY)
# 画点连线 (x1, x2, y1, y2)
print cur[0], cur[0]-delX, cur[1], cur[1]-delY
# print cur[0], cur[0]-delX, cur[1], cur[1]-delY
ax.plot([cur[0], cur[0]-delX], [cur[1], cur[1]-delY], c='b')
cur = (cur[0]-delX, cur[1]-delY)
# 画对角的虚线线