wassname/Castor
1
0
Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-09-09 11:13:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

added times to training

Browse Source
This commit is contained in:
Gaurav Baruah
2017-03-28 13:37:41 -04:00
parent ae77b6e57b
commit e6aa9bfe07
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sm-model/main.py
+2 -2
View File
@@ -126,7 +126,7 @@ if __name__ == "__main__":
if args.debugSingleBatch: sys.exit(0)
dev_accuracy, dev_scores = trainer.test(args.dataset_folder, 'clean-dev', args.batch_size, cache_file)
dev_map, dev_mrr = compute_map_mrr(args.dataset_folder, 'clean-dev', dev_scores)
logger.info("MAP {}, MRR {}".format(dev_map, dev_mrr))
logger.info("MAP {}\nMRR {}".format(dev_map, dev_mrr))
if np.fabs(dev_map - best_map) > 1e-3:
best_model = i
@@ -148,6 +148,6 @@ if __name__ == "__main__":
logger.info('Test set accuracy = {:.4f}'.format(test_accuracy))
map, mrr = compute_map_mrr(args.dataset_folder, 'clean-test', test_scores)
logger.info("MAP {}, MRR {}".format(map, mrr))
logger.info("MAP {}\nMRR {}".format(map, mrr))
sm-model/train.py
+3
View File
@@ -150,6 +150,8 @@ class Trainer(object):
def train(self, dataset_folder, set_folder, batch_size, word_vectors_cache_file, debugSingleBatch):
train_start_time = time.time()
# read in training data
questions, sentences, labels, vocab, maxlen_q, maxlen_s, ext_feats = \
utils.read_in_dataset(dataset_folder, set_folder)
@@ -193,6 +195,7 @@ class Trainer(object):
logger.info('train_loss = {:.4f}'.format(
train_loss/num_batches
))
logger.info('training time = {:.3f} seconds'.format(time.time() - train_start_time))
return train_correct/num_batches