MCPcopy Create free account
hub / github.com/Jack-Cherish/Machine-Learning / splitDataSet

Function splitDataSet

Decision Tree/Decision Tree.py:86–93  ·  view source on GitHub ↗
(dataSet, axis, value)

Source from the content-addressed store, hash-verified

84 2017-07-24
85"""
86def splitDataSet(dataSet, axis, value):
87 retDataSet = [] #创建返回的数据集列表
88 for featVec in dataSet: #遍历数据集
89 if featVec[axis] == value:
90 reducedFeatVec = featVec[:axis] #去掉axis特征
91 reducedFeatVec.extend(featVec[axis+1:]) #将符合条件的添加到返回的数据集
92 retDataSet.append(reducedFeatVec)
93 return retDataSet #返回划分后的数据集
94
95"""
96函数说明:选择最优特征

Callers 2

chooseBestFeatureToSplitFunction · 0.85
createTreeFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected