-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathconfig.yml
More file actions
38 lines (37 loc) · 865 Bytes
/
config.yml
File metadata and controls
38 lines (37 loc) · 865 Bytes
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
log_frequency: 1000
vocab_len: 36 # for purposes of one-hot encoding strings: the len([a-z]) + len([0-9])
batch_size: 16
submodules:
# random_noise:
# input_size: 1
# output_size: 16
# year_of_construction:
# output_size: 16
# registration_date:
# output_size: 16
# recorded_date:
# output_size: 16
# house_number:
# output_size: 16
# house_number_addition:
# output_size: 16
# purposes:
# output_size: 16
# postal_code:
# output_size: 16
geometry:
hidden_size: 64
output_size: 32 # output 'channels'
cnn_kernel_size: 5
maxpool_kernel_size: 3
late_fusion:
input_size: 16 # specify last dimension output size for all submodules to be able to concatenate
hidden_size: 256
output_size: 9
learning_rate: 1e-3
epochs: 20
data_loader:
batch_size: 32
shuffle: true
num_workers: 6
validation_size: 2000