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

Function binSplitDataSet

Regression Trees/regTrees.py:49–66  ·  view source on GitHub ↗

函数说明:根据特征切分数据集合 Parameters: dataSet - 数据集合 feature - 带切分的特征 value - 该特征的值 Returns: mat0 - 切分的数据集合0 mat1 - 切分的数据集合1 Website: http://www.cuijiahua.com/ Modify: 2017-12-12

(dataSet, feature, value)

Source from the content-addressed store, hash-verified

47 plt.show()
48
49def binSplitDataSet(dataSet, feature, value):
50 """
51 函数说明:根据特征切分数据集合
52 Parameters:
53 dataSet - 数据集合
54 feature - 带切分的特征
55 value - 该特征的值
56 Returns:
57 mat0 - 切分的数据集合0
58 mat1 - 切分的数据集合1
59 Website:
60 http://www.cuijiahua.com/
61 Modify:
62 2017-12-12
63 """
64 mat0 = dataSet[np.nonzero(dataSet[:,feature] > value)[0],:]
65 mat1 = dataSet[np.nonzero(dataSet[:,feature] <= value)[0],:]
66 return mat0, mat1
67
68def regLeaf(dataSet):
69 """

Callers 3

chooseBestSplitFunction · 0.85
createTreeFunction · 0.85
pruneFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected