from evaluation import evalrank, evalrank2,evalrank3,evalrank_vse,evalrank_maxpool,evalrank_f_c,evalrank_fanhua
import numpy as np
from transformers import BertTokenizer
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "7"

#evalrank("./runs/bert_adam_bcan_gpo_vseinfty_bcan/model_best.pth.tar", "../SCAN-master/data/", "test")
#evalrank2("../bcan_gpo/runs/bert_adam_bcan_gpo/model_best.pth.tar", "../SCAN-master/data/", "test")
#evalrank2("../bcan_gpo/runs/bert_adam_bcan_gpo2/model_best.pth.tar", "../SCAN-master/data/", "test")

#evalrank2("../bcan_gpo/runs/bert_adam_bcan_gpo/model_best.pth.tar", "../SCAN-master/data/", "test")

#evalrank("./runs/bert_adam_bcan_gpo_vseinfty_bcan_coco/model_best.pth.tar", "../SCAN-master/data/", "test")
evalrank_fanhua("./runs/bert_adam_bcan_gpo_vseinfty_bcan_coco/model_best.pth.tar", "../SCAN-master/data/", "test")
#evalrank3("../pycharmProject/runs/bigru_bcan_adam_mean_base2/model_best.pth.tar", "../SCAN-master/data/", "test")
#evalrank3("./runs/bigru_adam_bcan_mean2/model_best.pth.tar", "../SCAN-master/data/", "test")
#evalrank("./runs/bert_adam_bcan_gpo_shuffle_test/model_best.pth.tar", "../SCAN-master/data/", "shuffle_dev")
#evalrank("./runs/bert/model_best.pth.tar", "./data/", "test")