-
Notifications
You must be signed in to change notification settings - Fork 4
/
config.yaml
104 lines (104 loc) · 2.61 KB
/
config.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
version: twfr-gmm
description: Time-weighted Frequency Domain Audio Representation (TWFR) with GMM Estimator for Anomalous Sound Detection
# filepath
root_dir: ../../data/dcase2022dataset
train_dirs:
- ../../data/dcase2022dataset/dev_data/fan/train
- ../../data/dcase2022dataset/dev_data/slider/train
- ../../data/dcase2022dataset/dev_data/valve/train
- ../../data/dcase2022dataset/dev_data/ToyCar/train
- ../../data/dcase2022dataset/dev_data/ToyTrain/train
- ../../data/dcase2022dataset/dev_data/bearing/train
- ../../data/dcase2022dataset/dev_data/gearbox/train
add_train_dirs:
- ../../data/dcase2022dataset/eval_data/fan/train
- ../../data/dcase2022dataset/eval_data/slider/train
- ../../data/dcase2022dataset/eval_data/valve/train
- ../../data/dcase2022dataset/eval_data/ToyCar/train
- ../../data/dcase2022dataset/eval_data/ToyTrain/train
- ../../data/dcase2022dataset/eval_data/bearing/train
- ../../data/dcase2022dataset/eval_data/gearbox/train
valid_dirs:
- ../../data/dcase2022dataset/dev_data/fan/test
- ../../data/dcase2022dataset/dev_data/slider/test
- ../../data/dcase2022dataset/dev_data/valve/test
- ../../data/dcase2022dataset/dev_data/ToyCar/test
- ../../data/dcase2022dataset/dev_data/ToyTrain/test
- ../../data/dcase2022dataset/dev_data/bearing/test
- ../../data/dcase2022dataset/dev_data/gearbox/test
test_dirs:
- ../../data/eval_dataset/fan/test
- ../../data/eval_dataset/bearing/test
- ../../data/eval_dataset/slider/test
- ../../data/eval_dataset/valve/test
- ../../data/eval_dataset/ToyCar/test
- ../../data/eval_dataset/ToyTrain/test
- ../../data/eval_dataset/gearbox/test
# audio
sr: 16000
# train
seed: 526 # 41
decision_threshold: 0.9
save: True
pool_type: gwrp # mean, max, gwrp
# components of gmm
gmm_ns:
mean-gmm:
ToyCar: 2
ToyTrain: 2
bearing: 2
fan: 2
gearbox: 2
slider: 2
valve: 1
max-gmm:
ToyCar: 2
ToyTrain: 2
bearing: 2
fan: 2,
gearbox: 4
slider: 8
valve: 2
twfr-gmm:
ToyCar: 2
ToyTrain: 4
bearing: 2
fan: 2
gearbox: 4
slider: 1
valve: 2
smote-twfr-gmm:
ToyCar: 1
ToyTrain: 4
bearing: 2
fan: 2
gearbox: 2
slider: 1
valve: 2
# decay of gwrp
gwrp_decays:
twfr-gmm:
ToyCar: 1.00
ToyTrain: 0.90
bearing: 1.00
fan: 1.00
gearbox: 0.98
slider: 0.88
valve: 0.46
smote-twfr-gmm:
ToyCar: 1.00
ToyTrain: 0.90
bearing: 0.80
fan: 1.00
gearbox: 1.00
slider: 0.90
valve: 0.46
# use smote or not
smotes:
ToyCar: True
ToyTrain: False
bearing: True
fan: True
gearbox: True
slider: True
valve: False