11# ## Classification - Heterogeneously typed features (ints, floats, bools, strings)
22
3+ using Base. Test
34using DecisionTree
45
56m, n = 10 ^ 2 , 5 ;
@@ -11,10 +12,21 @@ labels = string.(tf[inds]);
1112features = Array {Any} (m, n);
1213features[:,:] = randn (m, n);
1314features[:,2 ] = string .(tf[randperm (m)]);
14- features[:,3 ] = round . (Int, features[:,3 ]);
15+ features[:,3 ] = round (Int, features[:,3 ]);
1516features[:,4 ] = tf[inds];
1617
17- build_tree (labels, features)
18- build_forest (labels, features,2 ,3 )
19- build_stump (labels, features)
18+ model = build_tree (labels, features);
19+ preds = apply_tree (model, features);
20+ cm = confusion_matrix (labels, preds);
21+ @test cm. accuracy > 0.7
22+
23+ model = build_forest (labels, features,2 ,3 );
24+ preds = apply_forest (model, features);
25+ cm = confusion_matrix (labels, preds);
26+ @test cm. accuracy > 0.7
27+
28+ model, coeffs = build_adaboost_stumps (labels, features, 7 );
29+ preds = apply_adaboost_stumps (model, coeffs, features);
30+ cm = confusion_matrix (labels, preds);
31+ @test cm. accuracy > 0.7
2032
0 commit comments