information_extraction

information_extraction#

get_dynamic_max_length(examples, default_max_length: int, dynamic_max_length: List[int]) int[source]#

get max_length by examples which you can change it by examples in batch

class UIETask(task, model, schema=None, **kwargs)[source]#

Bases: Task

Universal Information Extraction Task. :param task: The name of task. :type task: string :param model: The model name in the task. :type model: string :param kwargs: Additional keyword arguments passed along to the specific task. :type kwargs: dict, optional

class GPTask(task, model, **kwargs)[source]#

Bases: Task

Global Pointer for closed-domain information extraction Task. :param task: The name of task. :type task: string :param model: The model name in the task. :type model: string :param kwargs: Additional keyword arguments passed along to the specific task. :type kwargs: dict, optional