使用着名的Iris数据集与Julia决策树分类器,我得到以下树。
using RDatasets
using DecisionTree
iris = dataset("datasets", "iris")
features = convert(Array, iris[:, 1:4])
labels = convert(Array, iris[:, 5]);
model = build_tree(labels, features)
model = prune_tree(model, 0.9)
print_tree(model)
Feature 3, Threshold 3.0
L-> setosa : 50/50
R-> Feature 4, Threshold 1.8
L-> Feature 3, Threshold 5.0
L-> versicolor : 47/48
R-> Feature 4, Threshold 1.6
L-> virginica : 3/3
R-> Feature 1, Threshold 7.2
L-> versicolor : 2/2
R-> virginica : 1/1
R-> Feature 3, Threshold 4.9
L-> Feature 1, Threshold 6.0
L-> versicolor : 1/1
R-> virginica : 2/2
R-> virginica : 43/43
我无法解释一些分支之后的数字,例如" setosa:50/50"或" virginica:3/3"。
有人可以解释一下这些意思吗?
答案 0 :(得分:1)
看起来像节点" setosa:50/50" 50朵花被正确分类(50朵花变成了这个节点,50朵是setosa) versicolor:47/48意味着其中一个是virginica或setosa。