-
Notifications
You must be signed in to change notification settings - Fork 1
/
model_definition_5.yaml
54 lines (50 loc) · 1.23 KB
/
model_definition_5.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
#accuracy: 0.6366233766233766
#hits_at_k: 1.0
#loss: 0.601285001891
#overall_stats: { 'avg_f1_score_macro': 0.5941864648454791,
# 'avg_f1_score_micro': 0.6366233766233766,
# 'avg_f1_score_weighted': 0.5896871296208346,
input_features:
-
name: image_path
type: image
encoder: stacked_cnn
pool_stride : 2
preprocessing:
width: 64
height: 64
num_channels: 1
conv_layers:
-
num_filters: 16
filter_size: 2
pool_size: 2
dropout: true
-
num_filters: 32
filter_size: 2
pool_size: 2
dropout: true
-
num_filters: 16
filter_size: 2
pool_size: 2
dropout: true
output_features:
-
name: label
type: category
reduce_inputs: concat
fc_layers:
-
fc_size: 1024
dropout: true
-
fc_size: 256
dropout: true
-
fc_size: 64
dropout: true
training:
dropout_rate: 0.30
batch_size: 16