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
