about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/BNT/examples/static/dtree/test_zoo1.m
blob: 23c258b3f21aca4aa1c6f50f0f23dd5d4454232c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
% Here the training data is adapted from UCI ML repository, 'zoo' data

dtreeCPD=tree_CPD;

% load data
fname = fullfile(BNT_HOME, 'examples', 'static', 'uci_data', 'zoo', 'zoo1.data')
data=load(fname);
data=data';

data=transform_data_into_bnt_format(data, []);

% learn decision tree from data 
ns=2*ones(1,17);
ns(13)=6;
ns(17)=7;
dtreeCPD1=learn_params(dtreeCPD,1:17,data,ns,[],'stop_cases',5); % a node with less than 5 cases will not be splitted

% evaluate on data
[score,outputs]=evaluate_tree_performance(dtreeCPD1,1:17,data,ns,[]);
fprintf('Accuracy in old training data %6.3f\n',score);