/
dienen_classifier.yaml
75 lines (75 loc) · 1.94 KB
/
dienen_classifier.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
defaults:
gpu_id: 0
seed: 1234
(dataset)TrainGenerator:
class: BatchGenerator
in: (dataset)Partition->train
batch_task: (dataset)DataProcessor->out
x: !var batch_x
y: !var batch_y
batch_size: !var batch_size
extra_data:
statistics: (dataset)StatisticsTrain->out
cache: False
in_memory: !nocache True
seed: !var seed
(dataset)ValidationGenerator:
class: BatchGenerator
in: (dataset)Partition->validation
batch_task: (dataset)DataProcessor->out
x: !var batch_x
y: !var batch_y
batch_size: !var batch_size
extra_data:
statistics: (dataset)StatisticsValidation->out
cache: False
in_memory: !nocache True
shuffle: False
seed: !var seed
(dataset)TestGenerator:
class: BatchGenerator
in: (dataset)Partition->test
batch_task: (dataset)DataProcessor->out
x: !var batch_x
y: !var batch_y
batch_size: !var batch_size
extra_data:
statistics: (dataset)StatisticsTest->out
cache: False
in_memory: !nocache True
shuffle: False
seed: !var seed
(dataset)Model:
class: DienenModel
train_data: (dataset)TrainGenerator->out
validation_data: (dataset)ValidationGenerator->out
dienen_config: !var dienen_config
export: True
mods: !var (dataset)_dienen_mod
wandb_run: self->wandb
seed: !var seed
(dataset)Predict:
class: DienenPredict
data: (dataset)TestGenerator->out
model: (dataset)Model->out
export: True
return_column: filename
cache: False
(dataset)PredictionIndex:
class: Pool
in: (dataset)Predict->predictions
type: argmax
axis: -1
in_memory: True
(dataset)TargetIndex:
class: Pool
in: (dataset)Predict->targets
type: argmax
axis: -1
in_memory: True
(dataset)Metrics:
class: ClassificationMetrics
predictions: (dataset)PredictionIndex->out
targets: (dataset)TargetIndex->out
export: True
labels: (dataset)LabelEncode->labels