information_extraction_model

class UIE(encoding_model)

forward(input_ids, token_type_ids, pos_ids, att_mask)

Defines the computation performed at every call. Should be overridden by all subclasses.

Parameters

• *inputs (tuple) – unpacked tuple arguments

• **kwargs (dict) – unpacked dict arguments

class UIEM(encoding_model)

forward(input_ids, pos_ids)

Defines the computation performed at every call. Should be overridden by all subclasses.

Parameters

• *inputs (tuple) – unpacked tuple arguments

• **kwargs (dict) – unpacked dict arguments