args.server_ip = '' args.server_port = '' args.output_mode = "classification" args.save_model_steps = 2000 args.resume_epochs = 0 args.resume_steps = 0 # Important configurations args.data_dir = 'dataset/preprocessed' args.train_file = None args.dev_file = 'friends_majority_result.json' args.result_file = 'friends_all_result.json' args.train_batch_size = 32 args.eval_batch_size = 32 args.do_train = False args.do_eval = False args.do_run = True args.num_train_epochs = 1.0 args.max_seq_length = 256 args.processor = Others_OneSentence_Processor args.output_dir = None args.resume_dir = os.path.join(model_dir, 'friends_others/epoch_6') args.learning_rate = 1e-5 args.seed = 69847 args.included_labels = 7 trainer = Trainer(args) trainer.execute()
print "Training started" for n in range(1, trainSetSize+1): filename = "train%d.tif" % n print filename trainer.train(trainAnno, container, filename) trainRef = trainer.record(container) ############################################### print "Dev set recognition started" devRef = trainer.openAnnoFile("annotations/dev.txt") rightNum = 0.0 percentage = 0.0 for n in range(1, devSetSize+1): filename = "dev%d.tif" % n print filename predClass, probability = trainer.execute(trainRef, filename) actualClass = int(devRef[filename]) if actualClass == int(predClass): rightNum += 1 percentage = (rightNum / (devSetSize+1)) * 100 print "*** Recognition Result : ", str(percentage) + "% ***"