/
simple_molecules.yml
60 lines (45 loc) · 2.03 KB
/
simple_molecules.yml
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
# This file defines the default values for the MNIST task.
####################################################################
# 1. CONFIGURATION PARAMETERS that will be LOADED by the component.
####################################################################
# Folder where task will store data (LOADED)
data_folder: '~/data/simple-molecules'
# Defines the split that will be used used (LOADED)
# Options: training | validation | test
split: training
# Optional parameter (LOADED)
# When present, resizes the MNIST images from [28,28] to [width, height]
#resize_image: [height, width]
streams:
####################################################################
# 2. Keymappings associated with INPUT and OUTPUT streams.
####################################################################
# Stream containing batch of indices (OUTPUT)
# Every task MUST return that stream.
indices: indices
# Stream containing batch of images (OUTPUT)
images: images
# Stream containing targets (label ids) (OUTPUT)
targets: targets
# Stream containing labels (words) (OUTPUT)
labels: labels
globals:
####################################################################
# 3. Keymappings of variables that will be RETRIEVED from GLOBALS.
####################################################################
####################################################################
# 4. Keymappings associated with GLOBAL variables that will be SET.
####################################################################
# Width of the image (SET)
input_width: image_width
# Height of the image (SET)
input_height: image_height
# Depth of the image (SET)
input_depth: image_depth
# Number of output classes: 10 (SET)
num_classes: num_classes
# Label (word-idx) mappings (SET)
label_word_mappings: label_word_mappings
####################################################################
# 5. Keymappings associated with statistics that will be ADDED.
####################################################################