Modules¶

deepke.name_entity_re.multimodal.modules.dataset module¶

class deepke.name_entity_re.multimodal.modules.dataset.MMPNERProcessor(data_path, args)[source]¶

Bases: object

load_from_file(mode='train')[source]¶

class deepke.name_entity_re.multimodal.modules.dataset.MMPNERDataset(processor, label_mapping, img_path=None, aux_path=None, rcnn_img_path=None, max_seq=40, ignore_idx=- 100, aux_size=128, rcnn_size=64, mode='train')[source]¶: Bases: torch.utils.data.dataset.Dataset

deepke.name_entity_re.multimodal.modules.train module¶

class deepke.name_entity_re.multimodal.modules.train.Trainer(train_data=None, dev_data=None, test_data=None, model=None, process=None, label_map=None, args=None, logger=None, writer=None)[source]¶

Bases: object

train()[source]¶

evaluate(epoch)[source]¶

test(epoch)[source]¶

predict()[source]¶

multiModal_before_train()[source]¶