class GCNTrainer(object):
def __init__(self, args, emb_matrix=None):
self.args = args
self.emb_matrix = emb_matrix
self.model = GCNClassifier(args, emb_matrix=emb_matrix)
self.parameters = [
p for p in self.model.parameters() if p.requires_grad
]
self.model.cuda()
#self.optimizer = torch.optim.Adamax(self.parameters, lr=args.lr)
self.optimizer = torch.optim.Adam(self.parameters, lr=args.lr)
# load model
def load(self, filename):
try:
checkpoint = torch.load(filename)
except BaseException:
print("Cannot load model from {}".format(filename))
exit()
self.model.load_state_dict(checkpoint['model'])
self.args = checkpoint['config']
# save model
def save(self, filename):
params = {
'model': self.model.state_dict(),
'config': self.args,
}
try:
torch.save(params, filename)
print("model saved to {}".format(filename))
except BaseException:
print("[Warning: Saving failed... continuing anyway.]")
def update(self, batch):
batch = [b.cuda() for b in batch]
inputs = batch[:-1]
label = batch[-1]
# step forward
self.model.train()
self.optimizer.zero_grad()
logits, gcn_outputs = self.model(inputs)
loss = F.cross_entropy(logits, label, reduction='mean')
loss_by_class = F.cross_entropy(logits, label, reduction='none')
## Masks by Class
neg_mask = label.data == 0
neu_mask = label.data == 1
pos_mask = label.data == 2
neg_loss = torch.sum(neg_mask * loss_by_class).item()
neu_loss = torch.sum(neu_mask * loss_by_class).item()
pos_loss = torch.sum(pos_mask * loss_by_class).item()
corrects = (torch.max(logits, 1)[1].view(
label.size()).data == label.data).sum()
acc = 100.0 * np.float(corrects) / label.size()[0]
# backward
loss.backward()
self.optimizer.step()
return loss.data, acc, [neg_loss, neu_loss, pos_loss]
def predict(self, batch):
batch = [b.cuda() for b in batch]
inputs = batch[:-1]
label = batch[-1]
# forward
self.model.eval()
logits, gcn_outputs = self.model(inputs)
loss = F.cross_entropy(logits, label, reduction='mean')
loss_by_class = F.cross_entropy(logits, label, reduction='none')
## Masks by Class
neg_mask = label.data == 0
neu_mask = label.data == 1
pos_mask = label.data == 2
neg_loss = torch.sum(neg_mask * loss_by_class).item()
neu_loss = torch.sum(neu_mask * loss_by_class).item()
pos_loss = torch.sum(pos_mask * loss_by_class).item()
corrects = (torch.max(logits, 1)[1].view(
label.size()).data == label.data).sum()
acc = 100.0 * np.float(corrects) / label.size()[0]
predictions = np.argmax(logits.data.cpu().numpy(), axis=1).tolist()
predprob = F.softmax(logits, dim=1).data.cpu().numpy().tolist()
return loss.data, [neg_loss, neu_loss, pos_loss], \
acc, predictions, \
label.data.cpu().numpy().tolist(), predprob, \
gcn_outputs.data.cpu().numpy()
class GCNTrainer(object):
def __init__(self, args, emb_matrix=None):
self.args = args
self.emb_matrix = emb_matrix
self.model = GCNClassifier(args, emb_matrix=emb_matrix)
self.parameters = [
p for p in self.model.parameters() if p.requires_grad
]
self.model.cuda()
self.optimizer = torch_utils.get_optimizer(args.optim, self.parameters,
args.lr)
# load model_state and args
def load(self, filename):
try:
checkpoint = torch.load(filename)
except BaseException:
print("Cannot load model from {}".format(filename))
exit()
self.model.load_state_dict(checkpoint['model'])
self.args = checkpoint['config']
# save model_state and args
def save(self, filename):
params = {
'model': self.model.state_dict(),
'config': self.args,
}
try:
torch.save(params, filename)
print("model saved to {}".format(filename))
except BaseException:
print("[Warning: Saving failed... continuing anyway.]")
def update(self, batch):
# convert to cuda
batch = [b.cuda() for b in batch]
# unpack inputs and label
inputs = batch[0:8]
label = batch[-1]
# step forward
self.model.train()
self.optimizer.zero_grad()
logits, gcn_outputs = self.model(inputs)
loss = F.cross_entropy(logits, label, reduction='mean')
corrects = (torch.max(logits, 1)[1].view(
label.size()).data == label.data).sum()
acc = 100.0 * np.float(corrects) / label.size()[0]
# backward
loss.backward()
self.optimizer.step()
return loss.data, acc
def predict(self, batch):
# convert to cuda
batch = [b.cuda() for b in batch]
# unpack inputs and label
inputs = batch[0:8]
label = batch[-1]
# forward
self.model.eval()
logits, gcn_outputs = self.model(inputs)
loss = F.cross_entropy(logits, label, reduction='mean')
corrects = (torch.max(logits, 1)[1].view(
label.size()).data == label.data).sum()
acc = 100.0 * np.float(corrects) / label.size()[0]
predictions = np.argmax(logits.data.cpu().numpy(), axis=1).tolist()
predprob = F.softmax(logits, dim=1).data.cpu().numpy().tolist()
return loss.data, acc, predictions, label.data.cpu().numpy().tolist(
), predprob, gcn_outputs.data.cpu().numpy()