/
rnekrasov
/
DocRED
Обзор
Документация
Войти
/
rnekrasov
/
DocRED
Код
Запросы
0
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
code/test.py
40 строк
1 KB
YeDeming
initial commit
13 июн 2019, 17:10
13 июн 2019, 17:10
ee45b6a
Код
Авторство
О чём код?
import config import models import numpy as np import os import time import datetime import json from sklearn.metrics import average_precision_score import sys import os import argparse # import IPython # sys.excepthook = IPython.core.ultratb.FormattedTB(mode='Verbose', color_scheme='Linux', call_pdb=1) parser = argparse.ArgumentParser() parser.add_argument('--model_name', type = str, default = 'LSTM', help = 'name of the model') parser.add_argument('--save_name', type = str) parser.add_argument('--train_prefix', type = str, default = 'train') parser.add_argument('--test_prefix', type = str, default = 'dev_dev') parser.add_argument('--input_theta', type = float, default = -1) # parser.add_argument('--ignore_input_theta', type = float, default = -1) args = parser.parse_args() model = { 'CNN3': models.CNN3, 'LSTM': models.LSTM, 'BiLSTM': models.BiLSTM, 'ContextAware': models.ContextAware, # 'LSTM_SP': models.LSTM_SP } con = config.Config(args) #con.load_train_data() con.load_test_data() # con.set_train_model() con.testall(model[args.model_name], args.save_name, args.input_theta)#, args.ignore_input_theta)