def train_onehot(args, paths):
logger = Logger()
dset = load_classification_dataset(args.dataset)
t2i, i2t, in2i, i2in = dset['t2i'], dset['i2t'], dset['in2i'], dset['i2in']
query_train, intent_train = dset['query_train'], dset['intent_train']
query_dev, intent_dev = dset['query_dev'], dset['intent_dev']
query_test, intent_test = dset['query_test'], dset['intent_test']
len_stats(query_train)
len_stats(query_dev)
len_stats(query_test)
# extend the padding
# add pad <pad> to the last of vocab
i2t[len(i2t)] = '<pad>'
t2i['<pad>'] = len(i2t) - 1
train_query, train_query_inverse, train_lengths = pad_dataset(
query_train, args, t2i['<pad>'])
dev_query, dev_query_inverse, dev_lengths = pad_dataset(
query_dev, args, t2i['<pad>'])
test_query, test_query_inverse, test_lengths = pad_dataset(
query_test, args, t2i['<pad>'])
shots = int(len(train_query) * args.train_portion)
assert args.train_portion == 1.0
# We currently not support ublabel and low-resource for onehot
intent_data_train = ATISIntentBatchDataset(train_query, train_lengths,
intent_train, shots)
intent_data_dev = ATISIntentBatchDataset(dev_query, dev_lengths,
intent_dev, shots)
intent_data_test = ATISIntentBatchDataset(test_query, test_lengths,
intent_test)
intent_dataloader_train = DataLoader(intent_data_train, batch_size=args.bz)
intent_dataloader_dev = DataLoader(intent_data_dev, batch_size=args.bz)
intent_dataloader_test = DataLoader(intent_data_test, batch_size=args.bz)
automata_dicts = load_pkl(paths[0])
if 'automata' not in automata_dicts:
automata = automata_dicts
else:
automata = automata_dicts['automata']
language_tensor, state2idx, wildcard_mat, language = dfa_to_tensor(
automata, t2i)
complete_tensor = language_tensor + wildcard_mat
assert args.additional_state == 0
if args.dataset == 'ATIS':
mat, bias = create_mat_and_bias_with_empty_ATIS(
automata,
in2i=in2i,
i2in=i2in,
)
elif args.dataset == 'TREC':
mat, bias = create_mat_and_bias_with_empty_TREC(
automata,
in2i=in2i,
i2in=i2in,
)
elif args.dataset == 'SMS':
mat, bias = create_mat_and_bias_with_empty_SMS(
automata,
in2i=in2i,
i2in=i2in,
)
# for padding
V, S1, S2 = complete_tensor.shape
complete_tensor_extend = np.concatenate(
(complete_tensor, np.zeros((1, S1, S2))))
print(complete_tensor_extend.shape)
model = IntentIntegrateOnehot(complete_tensor_extend,
config=args,
mat=mat,
bias=bias)
mode = 'onehot'
if args.loss_type == 'CrossEntropy':
criterion = torch.nn.CrossEntropyLoss()
elif args.loss_type == 'NormalizeNLL':
criterion = relu_normalized_NLLLoss
else:
print("Wrong loss function")
if args.optimizer == 'SGD':
optimizer = torch.optim.SGD(model.parameters(),
lr=args.lr,
weight_decay=0)
if args.optimizer == 'ADAM':
optimizer = torch.optim.Adam(model.parameters(),
lr=args.lr,
weight_decay=0)
if torch.cuda.is_available():
model = model.cuda()
acc_train_init, avg_loss_train_init, p, r = val(model,
intent_dataloader_train,
epoch=0,
mode='TRAIN',
config=args,
i2in=i2in,
criterion=criterion)
# DEV
acc_dev_init, avg_loss_dev_init, p, r = val(model,
intent_dataloader_dev,
epoch=0,
mode='DEV',
config=args,
i2in=i2in,
criterion=criterion)
# TEST
acc_test_init, avg_loss_test_init, p, r = val(model,
intent_dataloader_test,
epoch=0,
mode='TEST',
config=args,
i2in=i2in,
criterion=criterion)
best_dev_acc = acc_dev_init
counter = 0
best_dev_test_acc = acc_test_init
for epoch in range(1, args.epoch + 1):
avg_loss = 0
acc = 0
pbar_train = tqdm(intent_dataloader_train)
pbar_train.set_description("TRAIN EPOCH {}".format(epoch))
model.train()
for batch in pbar_train:
optimizer.zero_grad()
x = batch['x']
label = batch['i'].view(-1)
lengths = batch['l']
if torch.cuda.is_available():
x = x.cuda()
lengths = lengths.cuda()
label = label.cuda()
scores = model(x, lengths)
loss_cross_entropy = criterion(scores, label)
loss = loss_cross_entropy
loss.backward()
optimizer.step()
avg_loss += loss.item()
acc += (scores.argmax(1) == label).sum().item()
pbar_train.set_postfix_str(
"{} - total right: {}, total loss: {}".format(
'TRAIN', acc, loss))
acc = acc / len(intent_data_train)
avg_loss = avg_loss / len(intent_data_train)
print("{} Epoch: {} | ACC: {}, LOSS: {}".format(
'TRAIN', epoch, acc, avg_loss))
logger.add("{} Epoch: {} | ACC: {}, LOSS: {}".format(
'TRAIN', epoch, acc, avg_loss))
# DEV
acc_dev, avg_loss_dev, p, r = val(model,
intent_dataloader_dev,
epoch,
'DEV',
logger,
config=args,
criterion=criterion)
# TEST
acc_test, avg_loss_test, p, r = val(model,
intent_dataloader_test,
epoch,
'TEST',
logger,
config=args,
criterion=criterion)
counter += 1 # counter for early stopping
if (acc_dev is None) or (acc_dev > best_dev_acc):
counter = 0
best_dev_acc = acc_dev
best_dev_test_acc = acc_test
if counter > args.early_stop:
break
return acc_dev_init, acc_test_init, best_dev_acc, best_dev_test_acc, logger.record
def get_init_params(config, in2i, i2in, t2i, automata_path):
dset = config.dataset
if 'SMS' in config.dataset:
dset = 'SMS'
pretrained_embed = load_glove_embed('../data/{}/'.format(dset),
config.embed_dim)
if config.random_embed:
pretrained_embed = np.random.random(pretrained_embed.shape)
automata_dicts = load_pkl(automata_path)
automata = automata_dicts['automata']
V_embed, D1, D2 = automata_dicts['V'], automata_dicts[
'D1'], automata_dicts['D2']
wildcard_mat, language = automata_dicts['wildcard_mat'], automata_dicts[
'language']
n_vocab, rank = V_embed.shape
n_state, _ = D1.shape
print("DFA states: {}".format(n_state))
_, embed_dim = pretrained_embed.shape
if dset == 'ATIS':
mat, bias = create_mat_and_bias_with_empty_ATIS(
automata,
in2i=in2i,
i2in=i2in,
)
elif dset == 'TREC':
mat, bias = create_mat_and_bias_with_empty_TREC(
automata,
in2i=in2i,
i2in=i2in,
)
elif dset == 'SMS':
mat, bias = create_mat_and_bias_with_empty_SMS(
automata,
in2i=in2i,
i2in=i2in,
)
# for padding
pretrain_embed_extend = np.append(pretrained_embed,
np.zeros((1, config.embed_dim),
dtype=np.float),
axis=0)
V_embed_extend = np.append(V_embed,
np.zeros((1, rank), dtype=np.float),
axis=0)
# creating language mask for regularization
n_vocab_extend, _ = V_embed_extend.shape
language_mask = torch.ones(n_vocab_extend)
language_mask[[t2i[i] for i in language]] = 0
# for V_embed_weighted mask and extend the wildcard mat to the right dimension
S, _ = wildcard_mat.shape
wildcard_mat_origin_extend = np.zeros(
(S + config.additional_state, S + config.additional_state))
wildcard_mat_origin_extend[:S, :S] = wildcard_mat
wildcard_mat_origin_extend = torch.from_numpy(
wildcard_mat_origin_extend).float()
if torch.cuda.is_available():
language_mask = language_mask.cuda()
wildcard_mat_origin_extend = wildcard_mat_origin_extend.cuda()
if config.normalize_automata != 'none':
D1_avg = get_average(D1, config.normalize_automata)
D2_avg = get_average(D2, config.normalize_automata)
V_embed_extend_avg = get_average(V_embed_extend,
config.normalize_automata)
factor = np.float_power(D1_avg * D2_avg * V_embed_extend_avg, 1 / 3)
print(factor)
print(D1_avg)
print(D2_avg)
print(V_embed_extend_avg)
D1 = D1 * (factor / D1_avg)
D2 = D2 * (factor / D2_avg)
V_embed_extend = V_embed_extend * (factor / V_embed_extend_avg)
return V_embed_extend, pretrain_embed_extend, mat, bias, D1, D2, language_mask, language, wildcard_mat, wildcard_mat_origin_extend
def PredictByRE(
args,
params=None,
dset=None,
):
logger = Logger()
if not dset:
dset = load_classification_dataset(args.dataset)
t2i, i2t, in2i, i2in = dset['t2i'], dset['i2t'], dset['in2i'], dset['i2in']
query_train, intent_train = dset['query_train'], dset['intent_train']
query_dev, intent_dev = dset['query_dev'], dset['intent_dev']
query_test, intent_test = dset['query_test'], dset['intent_test']
len_stats(query_train)
len_stats(query_dev)
len_stats(query_test)
# extend the padding
# add pad <pad> to the last of vocab
i2t[len(i2t)] = '<pad>'
t2i['<pad>'] = len(i2t) - 1
train_query, train_query_inverse, train_lengths = pad_dataset(
query_train, args, t2i['<pad>'])
dev_query, dev_query_inverse, dev_lengths = pad_dataset(
query_dev, args, t2i['<pad>'])
test_query, test_query_inverse, test_lengths = pad_dataset(
query_test, args, t2i['<pad>'])
intent_data_train = ATISIntentBatchDataset(train_query, train_lengths,
intent_train)
intent_data_dev = ATISIntentBatchDataset(dev_query, dev_lengths,
intent_dev)
intent_data_test = ATISIntentBatchDataset(test_query, test_lengths,
intent_test)
intent_dataloader_train = DataLoader(intent_data_train, batch_size=args.bz)
intent_dataloader_dev = DataLoader(intent_data_dev, batch_size=args.bz)
intent_dataloader_test = DataLoader(intent_data_test, batch_size=args.bz)
if params is None:
automata_dicts = load_pkl(args.automata_path_forward)
automata = automata_dicts['automata']
language_tensor, state2idx, wildcard_mat, language = dfa_to_tensor(
automata, t2i)
complete_tensor = language_tensor + wildcard_mat
if args.dataset == 'ATIS':
mat, bias = create_mat_and_bias_with_empty_ATIS(
automata,
in2i=in2i,
i2in=i2in,
)
elif args.dataset == 'TREC':
mat, bias = create_mat_and_bias_with_empty_TREC(
automata,
in2i=in2i,
i2in=i2in,
)
elif args.dataset == 'SMS':
mat, bias = create_mat_and_bias_with_empty_SMS(
automata,
in2i=in2i,
i2in=i2in,
)
else:
complete_tensor = params['complete_tensor']
mat, bias = params['mat'], params['bias']
# for padding
V, S1, S2 = complete_tensor.shape
complete_tensor_extend = np.concatenate(
(complete_tensor, np.zeros((1, S1, S2))))
print(complete_tensor_extend.shape)
model = IntentIntegrateOnehot(complete_tensor_extend,
config=args,
mat=mat,
bias=bias)
if torch.cuda.is_available():
model.cuda()
# # TRAIN
print('RE TRAIN ACC')
all_pred_train, all_out_train = REclassifier(model,
intent_dataloader_train,
config=args,
i2in=i2in)
# DEV
print('RE DEV ACC')
all_pred_dev, all_out_dev = REclassifier(model,
intent_dataloader_dev,
config=args,
i2in=i2in)
# TEST
print('RE TEST ACC')
all_pred_test, all_out_test = REclassifier(model,
intent_dataloader_test,
config=args,
i2in=i2in)
return all_pred_train, all_pred_dev, all_pred_test, all_out_train, all_out_dev, all_out_test