def __init__(self, agents, judger, args): assert len(agents) == 2 self.agents = agents self.judger = judger self.args = args self.domain = domain.get_domain(args.domain) self.metrics = MetricsContainer() self._register_metrics()
def __init__(self, dictionary, corpus, np2var, bad_tokens=['<disconnect>', '<disagree>']): super(CombinedNLLEntropy4CLF, self).__init__() self.dictionary = dictionary self.domain = domain.get_domain('object_division') self.corpus = corpus self.np2var = np2var self.bad_tokens = bad_tokens
def __init__(self, model, corpus, args, name): super(LstmAgent, self).__init__() self.model = model self.corpus = corpus self.args = args self.name = name self.domain = domain.get_domain(args.domain) self.context = None self.goal_h = None self.dlg_history = None # self.lang_is = None self.lang_os = None self.lang_h = None
from latent_dialog.utils import Pack, set_seed, prepare_dirs_loggers from latent_dialog.corpora import DealCorpus from latent_dialog.data_loaders import DealDataLoaders from latent_dialog.evaluators import BleuEvaluator from latent_dialog.models_deal import GaussHRED from latent_dialog.main import train, validate, generate import latent_dialog.domain as domain stats_path = 'config_log_model' if not os.path.exists(stats_path): os.mkdir(stats_path) start_time = time.strftime('%Y-%m-%d-%H-%M-%S', time.localtime(time.time())) print('[START]', start_time, '='*30) domain_name = 'object_division' domain_info = domain.get_domain(domain_name) config = Pack( train_path = '../data/negotiate/train.txt', val_path = '../data/negotiate/val.txt', test_path = '../data/negotiate/test.txt', last_n_model = 5, max_utt_len = 20, backward_size = 14, batch_size = 32, grad_clip=3.0, use_gpu = True, op = 'adam', init_lr = 0.001, l2_norm=0.00001, momentum = 0.0,
def __init__(self, agents, judger, args): assert len(agents) == 2 self.agents = agents self.judger = judger self.args = args self.domain = domain.get_domain(args.domain)
def __init__(self, model, device_id): self.model = model self.domain = domain.get_domain('object_division') self.device_id = device_id