Modules

deepke.relation_extraction.multimodal.modules.dataset module

class deepke.relation_extraction.multimodal.modules.dataset.MMREProcessor(data_path, re_path, args)

Bases: object

load_from_file(mode='train')

get_relation_dict()

get_rel2id(train_path)

class deepke.relation_extraction.multimodal.modules.dataset.MMREDataset(processor, transform, img_path=None, aux_img_path=None, mode='train')

Bases: torch.utils.data.dataset.Dataset

deepke.relation_extraction.multimodal.modules.metrics module

deepke.relation_extraction.multimodal.modules.metrics.eval_result(true_labels, pred_result, rel2id, logger, use_name=False)

deepke.relation_extraction.multimodal.modules.train module

class deepke.relation_extraction.multimodal.modules.train.Trainer(train_data=None, dev_data=None, test_data=None, re_dict=None, model=None, args=None, logger=None, writer=None)

Bases: object

train()

evaluate(epoch)

test(epoch)

predict()

before_multimodal_train()