python中value的含义_生成的scikit学习决策树中的value属性的含义是什么?

运行此代码并通过graphviz生成树图像后,我们可以观察到树中每个节点上都有值数据。在import pandas as pd

from sklearn.tree import DecisionTreeClassifier, export_graphviz

dta = pd.read_csv("data/adult.data.cleaned.csv")

for col in dta:

if not dta[col].dtype.kind == "O":

continue

if dta[col].str.contains("\?").any():

dta.ix[dta[col].str.contains("\?"), col] = "Other"

test.ix[test[col].str.contains("\?"), col] = "Other"

dta.income.replace({"<=50K": 0, ">50K": 1}, inplace=True)

test.income.replace({"<=50K": 0, ">50K": 1}, inplace=True)

y = dta.pop("income")

y_test = test.pop("income")

X_train = pd.get_dummies(dta)

X_test = pd.get_dummies(test)

X_test[X_train.columns.difference(X_test.columns)[0]] = 0

dtree = DecisionTreeClassifier(criterion='entropy', random_state=0, max_depth=6)

dtree.fit(X_train, y)

export_graphviz(dtree, feature_names=X_train.columns)

值属性代表什么?

EDIT:意味着在每个节点中都有一个value=[x, y]属性

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值