Skip to content

Commit a9e157e

Browse files
committed
no ref
1 parent 91287c1 commit a9e157e

File tree

4 files changed

+112
-133
lines changed

4 files changed

+112
-133
lines changed

src/BA-Trees/src/temp_ba_trees/forest.txt

Lines changed: 97 additions & 67 deletions
Original file line numberDiff line numberDiff line change
@@ -2,108 +2,138 @@ DATASET_NAME: dataset.train.csv
22
ENSEMBLE: RF
33
NB_TREES: 10
44
NB_FEATURES: 6
5-
NB_CLASSES: 2
5+
NB_CLASSES: 4
66
MAX_TREE_DEPTH: 3
77
Format: node / node type (LN - leaf node, IN - internal node) left child / right child / feature / threshold / node_depth / majority class (starts with index 0)
88

99
[TREE 1]
1010
NB_NODES: 9
11-
0 IN 1 4 5 0.5 0 -1
12-
1 IN 2 3 1 0.5 1 -1
13-
2 LN -1 -1 -1 -1 2 0
14-
3 LN -1 -1 -1 -1 2 1
15-
4 IN 5 6 4 0.5 1 -1
16-
5 LN -1 -1 -1 -1 2 1
17-
6 IN 7 8 0 37.4 2 -1
18-
7 LN -1 -1 -1 -1 3 0
19-
8 LN -1 -1 -1 -1 3 1
11+
0 IN 1 2 5 1.5 0 -1
12+
1 LN -1 -1 -1 -1 1 0
13+
2 IN 3 6 4 1.5 1 -1
14+
3 IN 4 5 5 2.5 2 -1
15+
4 LN -1 -1 -1 -1 3 0
16+
5 LN -1 -1 -1 -1 3 0
17+
6 IN 7 8 1 2.5 2 -1
18+
7 LN -1 -1 -1 -1 3 1
19+
8 LN -1 -1 -1 -1 3 0
2020

2121
[TREE 2]
22-
NB_NODES: 7
23-
0 IN 1 2 2 0.5 0 -1
24-
1 LN -1 -1 -1 -1 1 0
25-
2 IN 3 6 1 0.5 1 -1
26-
3 IN 4 5 5 0.5 2 -1
22+
NB_NODES: 13
23+
0 IN 1 8 0 2.5 0 -1
24+
1 IN 2 5 4 1.5 1 -1
25+
2 IN 3 4 3 3.0 2 -1
26+
3 LN -1 -1 -1 -1 3 0
2727
4 LN -1 -1 -1 -1 3 0
28-
5 LN -1 -1 -1 -1 3 1
29-
6 LN -1 -1 -1 -1 2 1
28+
5 IN 6 7 1 2.5 2 -1
29+
6 LN -1 -1 -1 -1 3 0
30+
7 LN -1 -1 -1 -1 3 0
31+
8 IN 9 10 5 1.5 1 -1
32+
9 LN -1 -1 -1 -1 2 0
33+
10 IN 11 12 1 2.5 2 -1
34+
11 LN -1 -1 -1 -1 3 1
35+
12 LN -1 -1 -1 -1 3 0
3036

3137
[TREE 3]
32-
NB_NODES: 7
33-
0 IN 1 4 5 0.5 0 -1
34-
1 IN 2 3 1 0.5 1 -1
35-
2 LN -1 -1 -1 -1 2 0
36-
3 LN -1 -1 -1 -1 2 1
37-
4 IN 5 6 2 0.5 1 -1
38-
5 LN -1 -1 -1 -1 2 0
39-
6 LN -1 -1 -1 -1 2 1
38+
NB_NODES: 9
39+
0 IN 1 2 3 3.0 0 -1
40+
1 LN -1 -1 -1 -1 1 0
41+
2 IN 3 6 1 2.5 1 -1
42+
3 IN 4 5 1 1.5 2 -1
43+
4 LN -1 -1 -1 -1 3 0
44+
5 LN -1 -1 -1 -1 3 0
45+
6 IN 7 8 0 2.5 2 -1
46+
7 LN -1 -1 -1 -1 3 0
47+
8 LN -1 -1 -1 -1 3 0
4048

4149
[TREE 4]
4250
NB_NODES: 11
43-
0 IN 1 4 3 0.5 0 -1
44-
1 IN 2 3 0 38.95 1 -1
51+
0 IN 1 6 1 2.5 0 -1
52+
1 IN 2 3 5 1.5 1 -1
4553
2 LN -1 -1 -1 -1 2 0
46-
3 LN -1 -1 -1 -1 2 1
47-
4 IN 5 8 4 0.5 1 -1
48-
5 IN 6 7 2 0.5 2 -1
49-
6 LN -1 -1 -1 -1 3 0
50-
7 LN -1 -1 -1 -1 3 1
51-
8 IN 9 10 1 0.5 2 -1
54+
3 IN 4 5 0 2.5 2 -1
55+
4 LN -1 -1 -1 -1 3 0
56+
5 LN -1 -1 -1 -1 3 0
57+
6 IN 7 8 5 1.5 1 -1
58+
7 LN -1 -1 -1 -1 2 0
59+
8 IN 9 10 3 3.0 2 -1
5260
9 LN -1 -1 -1 -1 3 0
53-
10 LN -1 -1 -1 -1 3 1
61+
10 LN -1 -1 -1 -1 3 0
5462

5563
[TREE 5]
5664
NB_NODES: 7
57-
0 IN 1 6 1 0.5 0 -1
58-
1 IN 2 3 2 0.5 1 -1
59-
2 LN -1 -1 -1 -1 2 0
60-
3 IN 4 5 3 0.5 2 -1
61-
4 LN -1 -1 -1 -1 3 0
65+
0 IN 1 2 5 1.5 0 -1
66+
1 LN -1 -1 -1 -1 1 0
67+
2 IN 3 4 3 3.0 1 -1
68+
3 LN -1 -1 -1 -1 2 0
69+
4 IN 5 6 0 2.5 2 -1
6270
5 LN -1 -1 -1 -1 3 1
63-
6 LN -1 -1 -1 -1 1 1
71+
6 LN -1 -1 -1 -1 3 0
6472

6573
[TREE 6]
66-
NB_NODES: 5
67-
0 IN 1 2 2 0.5 0 -1
74+
NB_NODES: 9
75+
0 IN 1 2 5 1.5 0 -1
6876
1 LN -1 -1 -1 -1 1 0
69-
2 IN 3 4 0 37.85 1 -1
70-
3 LN -1 -1 -1 -1 2 0
71-
4 LN -1 -1 -1 -1 2 1
77+
2 IN 3 6 0 2.5 1 -1
78+
3 IN 4 5 4 1.5 2 -1
79+
4 LN -1 -1 -1 -1 3 0
80+
5 LN -1 -1 -1 -1 3 1
81+
6 IN 7 8 3 3.0 2 -1
82+
7 LN -1 -1 -1 -1 3 0
83+
8 LN -1 -1 -1 -1 3 1
7284

7385
[TREE 7]
74-
NB_NODES: 7
75-
0 IN 1 6 1 0.5 0 -1
76-
1 IN 2 3 2 0.5 1 -1
86+
NB_NODES: 13
87+
0 IN 1 6 1 2.5 0 -1
88+
1 IN 2 3 3 3.0 1 -1
7789
2 LN -1 -1 -1 -1 2 0
78-
3 IN 4 5 0 37.75 2 -1
90+
3 IN 4 5 5 1.5 2 -1
7991
4 LN -1 -1 -1 -1 3 0
8092
5 LN -1 -1 -1 -1 3 1
81-
6 LN -1 -1 -1 -1 1 1
93+
6 IN 7 10 4 1.5 1 -1
94+
7 IN 8 9 5 2.5 2 -1
95+
8 LN -1 -1 -1 -1 3 0
96+
9 LN -1 -1 -1 -1 3 0
97+
10 IN 11 12 5 1.5 2 -1
98+
11 LN -1 -1 -1 -1 3 0
99+
12 LN -1 -1 -1 -1 3 0
82100

83101
[TREE 8]
84-
NB_NODES: 5
85-
0 IN 1 2 0 37.95 0 -1
102+
NB_NODES: 7
103+
0 IN 1 2 3 3.0 0 -1
86104
1 LN -1 -1 -1 -1 1 0
87-
2 IN 3 4 2 0.5 1 -1
105+
2 IN 3 4 5 1.5 1 -1
88106
3 LN -1 -1 -1 -1 2 0
89-
4 LN -1 -1 -1 -1 2 1
107+
4 IN 5 6 0 2.5 2 -1
108+
5 LN -1 -1 -1 -1 3 1
109+
6 LN -1 -1 -1 -1 3 0
90110

91111
[TREE 9]
92-
NB_NODES: 7
93-
0 IN 1 2 0 37.95 0 -1
94-
1 LN -1 -1 -1 -1 1 0
95-
2 IN 3 6 3 0.5 1 -1
96-
3 IN 4 5 4 0.5 2 -1
112+
NB_NODES: 11
113+
0 IN 1 6 0 2.5 0 -1
114+
1 IN 2 3 5 1.5 1 -1
115+
2 LN -1 -1 -1 -1 2 0
116+
3 IN 4 5 3 3.0 2 -1
97117
4 LN -1 -1 -1 -1 3 0
98118
5 LN -1 -1 -1 -1 3 1
99-
6 LN -1 -1 -1 -1 2 1
119+
6 IN 7 10 1 3.5 1 -1
120+
7 IN 8 9 3 3.0 2 -1
121+
8 LN -1 -1 -1 -1 3 0
122+
9 LN -1 -1 -1 -1 3 0
123+
10 LN -1 -1 -1 -1 2 0
100124

101125
[TREE 10]
102-
NB_NODES: 7
103-
0 IN 1 2 0 37.95 0 -1
104-
1 LN -1 -1 -1 -1 1 0
105-
2 IN 3 6 3 0.5 1 -1
106-
3 IN 4 5 2 0.5 2 -1
126+
NB_NODES: 13
127+
0 IN 1 8 0 2.5 0 -1
128+
1 IN 2 5 0 1.5 1 -1
129+
2 IN 3 4 5 1.5 2 -1
130+
3 LN -1 -1 -1 -1 3 0
107131
4 LN -1 -1 -1 -1 3 0
108-
5 LN -1 -1 -1 -1 3 1
109-
6 LN -1 -1 -1 -1 2 1
132+
5 IN 6 7 2 2.5 2 -1
133+
6 LN -1 -1 -1 -1 3 0
134+
7 LN -1 -1 -1 -1 3 0
135+
8 IN 9 10 5 1.5 1 -1
136+
9 LN -1 -1 -1 -1 2 0
137+
10 IN 11 12 1 2.5 2 -1
138+
11 LN -1 -1 -1 -1 3 1
139+
12 LN -1 -1 -1 -1 3 0
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
dataset.train.csv,RF,10,6,2,0,6,29,30,1,0.000748,192,1725,5248
1+
dataset.train.csv,RF,10,6,4,0,5,5,6,1,0.000159,32,152,395

src/BA-Trees/src/temp_ba_trees/result.txt.tree

Lines changed: 14 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -2,68 +2,20 @@ DATASET_NAME: dataset.train.csv
22
ENSEMBLE: BA
33
NB_TREES: 1
44
NB_FEATURES: 6
5-
NB_CLASSES: 2
6-
MAX_TREE_DEPTH: 6
5+
NB_CLASSES: 4
6+
MAX_TREE_DEPTH: 5
77
Format: node / node type(LN - leave node, IN - internal node) left child / right child / feature / threshold / node_depth / majority class (starts with index 0)
88

99
[TREE 0]
10-
NB_NODES: 59
11-
0 IN 1 30 1 0.5 0 -1
12-
1 IN 2 3 2 0.5 1 -1
13-
2 LN -1 -1 -1 -1 2 0
14-
3 IN 4 15 3 0.5 2 -1
15-
4 IN 5 10 0 38.95 3 -1
16-
5 IN 6 7 0 37.95 4 -1
17-
6 LN -1 -1 -1 -1 5 0
18-
7 IN 8 9 5 0.5 5 -1
19-
8 LN -1 -1 -1 -1 6 0
20-
9 LN -1 -1 -1 -1 6 1
21-
10 IN 11 14 4 0.5 4 -1
22-
11 IN 12 13 5 0.5 5 -1
23-
12 LN -1 -1 -1 -1 6 0
24-
13 LN -1 -1 -1 -1 6 1
25-
14 LN -1 -1 -1 -1 5 1
26-
15 IN 16 23 4 0.5 3 -1
27-
16 IN 17 20 5 0.5 4 -1
28-
17 IN 18 19 0 37.95 5 -1
29-
18 LN -1 -1 -1 -1 6 0
30-
19 LN -1 -1 -1 -1 6 1
31-
20 IN 21 22 0 37.75 5 -1
32-
21 LN -1 -1 -1 -1 6 0
33-
22 LN -1 -1 -1 -1 6 1
34-
23 IN 24 27 5 0.5 4 -1
35-
24 IN 25 26 0 37.95 5 -1
36-
25 LN -1 -1 -1 -1 6 0
37-
26 LN -1 -1 -1 -1 6 1
38-
27 IN 28 29 0 37.85 5 -1
39-
28 LN -1 -1 -1 -1 6 0
40-
29 LN -1 -1 -1 -1 6 1
41-
30 IN 31 46 2 0.5 1 -1
42-
31 IN 32 39 3 0.5 2 -1
43-
32 IN 33 34 0 38.95 3 -1
44-
33 LN -1 -1 -1 -1 4 0
45-
34 IN 35 36 4 0.5 4 -1
46-
35 LN -1 -1 -1 -1 5 0
47-
36 IN 37 38 5 0.5 5 -1
48-
37 LN -1 -1 -1 -1 6 1
49-
38 LN -1 -1 -1 -1 6 0
50-
39 IN 40 41 0 37.95 3 -1
51-
40 LN -1 -1 -1 -1 4 0
52-
41 IN 42 45 4 0.5 4 -1
53-
42 IN 43 44 5 0.5 5 -1
54-
43 LN -1 -1 -1 -1 6 1
55-
44 LN -1 -1 -1 -1 6 0
56-
45 LN -1 -1 -1 -1 5 1
57-
46 IN 47 54 0 37.4 2 -1
58-
47 IN 48 49 3 0.5 3 -1
59-
48 LN -1 -1 -1 -1 4 0
60-
49 IN 50 51 4 0.5 4 -1
61-
50 LN -1 -1 -1 -1 5 1
62-
51 IN 52 53 5 0.5 5 -1
63-
52 LN -1 -1 -1 -1 6 1
64-
53 LN -1 -1 -1 -1 6 0
65-
54 IN 55 58 0 37.85 3 -1
66-
55 IN 56 57 3 0.5 4 -1
67-
56 LN -1 -1 -1 -1 5 0
68-
57 LN -1 -1 -1 -1 5 1
69-
58 LN -1 -1 -1 -1 4 1
10+
NB_NODES: 11
11+
0 IN 1 10 0 2.5 0 -1
12+
1 IN 2 9 1 2.5 1 -1
13+
2 IN 3 4 3 3 2 -1
14+
3 LN -1 -1 -1 -1 3 0
15+
4 IN 5 6 4 1.5 3 -1
16+
5 LN -1 -1 -1 -1 4 0
17+
6 IN 7 8 5 1.5 4 -1
18+
7 LN -1 -1 -1 -1 5 0
19+
8 LN -1 -1 -1 -1 5 1
20+
9 LN -1 -1 -1 -1 2 0
21+
10 LN -1 -1 -1 -1 1 0

src/lumen/main.jl

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,9 +89,6 @@ No explicit return value, but produces:
8989
- The function aligns with Algorithm 1 from the reference implementation
9090
- Performance statistics are generated based on processing time and rule reduction
9191
92-
# References
93-
- Pagliarini, Giovanni, et al. "Minimal Rules from Decision Forests: a Systematic Approach." OVERLAY@ AI* IA. 2024.
94-
9592
# Example
9693
```julia
9794
model = load_decision_tree_model()

0 commit comments

Comments
 (0)