information_extraction_model

class UIE(encoding_model)[source]
forward(input_ids, token_type_ids, pos_ids, att_mask)[source]

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)[source]
forward(input_ids, pos_ids)[source]

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

Parameters
  • *inputs (tuple) – unpacked tuple arguments

  • **kwargs (dict) – unpacked dict arguments