configs/vqa_med_2019/question_categorization/question_categorization_rnn.yml
# Load config defining tasks for training, validation and testing.
default_configs: vqa_med_2019/question_categorization/default_question_categorization.yml
pipeline:
# Questions encoding.
question_tokenizer:
type: SentenceTokenizer
priority: 1.1
streams:
inputs: questions
outputs: tokenized_questions
# Model 2: Embeddings
question_embeddings:
type: SentenceEmbeddings
priority: 1.2
embeddings_size: 50
pretrained_embeddings_file: glove.6B.50d.txt
data_folder: ~/data/vqa-med
word_mappings_file: questions.all.word.mappings.csv
streams:
inputs: tokenized_questions
outputs: embedded_questions
# Model 2: RNN
lstm:
type: RecurrentNeuralNetwork
cell_type: LSTM
prediction_mode: Last
priority: 3
initial_state: Zero
streams:
inputs: embedded_questions
globals:
input_size: embeddings_size
prediction_size: num_categories # C1,C2,C3,C4,BIN,UNK
#: pipeline