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.

参数
  • *inputs (tuple) -- unpacked tuple arguments

  • **kwargs (dict) -- unpacked dict arguments