We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
wrap
1 parent 42fd264 commit d7fccfaCopy full SHA for d7fccfa
test/miscellaneous/abstract_trees_test.jl
@@ -81,4 +81,23 @@ end
81
traverse_tree(leaf::InfoLeaf) = nothing
82
83
traverse_tree(wrapped_tree)
84
+end
85
+
86
+@testset "abstract_trees - test misuse" begin
87
88
+ @info("Test misuse of `classlabel` information")
89
90
+ @info("Create test data - a decision tree based on the iris data set")
91
+ features, labels = load_data("iris")
92
+ features = float.(features)
93
+ labels = string.(labels)
94
+ model = DecisionTreeClassifier()
95
+ fit!(model, features, labels)
96
97
+ @info("Try to replace the exisitng class labels")
98
+ class_labels = unique(labels)
99
+ dtree = model.root.node
100
+ wt = DecisionTree.wrap(dtree, (classlabels = class_labels,))
101
+ @test_throws AssertionError AbstractTrees.print_tree(wt)
102
103
end
0 commit comments