subject_ids, corpus, chart_dates = db.get_corpus(
toy_set=args.toy_set,
top100_labels=args.top100_labels,
validation_set=args.validation_set,
test_set=args.test_set)
bag_of_words_generator = BagOfWordsGenerator(logger, vocabulary,
subject_ids, corpus,
chart_dates)
if args.for_rnn:
bag_of_words_vectors_rnn = bag_of_words_generator.build_bag_of_words_vectors_rnn(
)
logger.info('Bag of words vectors for RNN created')
db.insert_bag_of_words_vectors_rnn(bag_of_words_vectors_rnn,
table_name)
logger.info('Bag of words vectors for RNN inserted in table %s',
table_name)
else:
bag_of_words_vectors = bag_of_words_generator.build_bag_of_words_vectors(
)
logger.info('Bag of words vectors created')
db.insert_bag_of_words_vectors(bag_of_words_vectors, table_name)
logger.info('Bag of words vectors inserted in table %s', table_name)
end = datetime.datetime.now()
db.bag_of_words_generator_experiment_insert_table_name(
experiment_id, end, table_name)
logger.info('Bag of words vectors inserted into database, table = %s',
table_name)
print(table_name)
