-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathensembleconfig.yaml
106 lines (97 loc) · 2.01 KB
/
ensembleconfig.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
# models:
# - name: Bar
# input_channels: 3
# num_classes: 5
# model_type: "Simple2DCNN"
# chart_type: "bar_border_monochrome_with_label"
# apply_smote: false
# smote_params:
# 1: 200
# 2: 200
# 3: 250
# 4: 300
# - name: Area
# input_channels: 3
# num_classes: 5
# model_type: "Simple2DCNN"
# chart_type: "area_monochrome_without_label"
# apply_smote: false
# smote_params:
# 1: 200
# 2: 200
# 3: 250
# 4: 300
# - name: Line
# input_channels: 3
# num_classes: 5
# model_type: "Deep2DCNN"
# chart_type: "line_color_without_label"
# apply_smote: false
# smote_params:
# 1: 200
# 2: 200
# 3: 250
# 4: 300
# - name: Scatter
# input_channels: 3
# num_classes: 5
# model_type: "Deep2DCNN"
# chart_type: "scatter_plain_color_without_label"
# apply_smote: true
# smote_params:
# 1: 200
# 2: 200
# 3: 250
# 4: 300
# ensemble_strategy:
# method: "majority_vote"
# num_models: 4
# Best settings
models:
- name: Bar
input_channels: 3
num_classes: 5
model_type: "Deep2DCNN"
chart_type: "bar_border_color_without_label"
apply_smote: true
smote_params:
1: 200
2: 200
3: 250
4: 300
- name: Area
input_channels: 3
num_classes: 5
model_type: "Deep2DCNN"
chart_type: "area_color_without_label"
apply_smote: true
smote_params:
1: 200
2: 200
3: 250
4: 300
- name: Line
input_channels: 3
num_classes: 5
model_type: "Deep2DCNN"
chart_type: "line_color_without_label"
apply_smote: true
smote_params:
1: 200
2: 200
3: 250
4: 300
- name: Scatter
input_channels: 3
num_classes: 5
model_type: "Deep2DCNN"
chart_type: "scatter_plain_color_without_label"
apply_smote: true
smote_params:
1: 200
2: 200
3: 250
4: 300
ensemble_strategy:
method: "majority_vote"
num_models: 4