๐Ÿ“„ Zero Or Oneยถ

This is binary text classification task where the labels are provided as 0 and 1.

The โ€˜classโ€™ is the name of the class we classifify and must be the same in all instances. The โ€˜text_typeโ€™ is an optional field that defines the type of text we classify (e.g. โ€œdocumentโ€, โ€œreviewโ€, etc.). This can be used by the template to customize the prompt.

The default reported metrics are the classifical f1_micro (accuracy).

tasks.classification.binary.zero_or_one

type: Task
input_fields: 
  text: str
  text_type: str
  class: str
reference_fields: 
  class: str
  label: int
prediction_type: float
metrics: 
  - metrics.accuracy
  - metrics.f1_binary
augmentable_inputs: 
  - text
defaults: 
  text_type: text
[source]

Explanation about Taskยถ

Task packs the different instance fields into dictionaries by their roles in the task.

Args:
input_fields (Union[Dict[str, str], List[str]]):

Dictionary with string names of instance input fields and types of respective values. In case a list is passed, each type will be assumed to be Any.

reference_fields (Union[Dict[str, str], List[str]]):

Dictionary with string names of instance output fields and types of respective values. In case a list is passed, each type will be assumed to be Any.

metrics (List[str]):

List of names of metrics to be used in the task.

prediction_type (Optional[str]):

Need to be consistent with all used metrics. Defaults to None, which means that it will be set to Any.

defaults (Optional[Dict[str, Any]]):

An optional dictionary with default values for chosen input/output keys. Needs to be consistent with names and types provided in โ€˜input_fieldsโ€™ and/or โ€˜output_fieldsโ€™ arguments. Will not overwrite values if already provided in a given instance.

The output instance contains three fields:
  1. โ€œinput_fieldsโ€ whose value is a sub-dictionary of the input instance, consisting of all the fields listed in Arg โ€˜input_fieldsโ€™.

  2. โ€œreference_fieldsโ€ โ€“ for the fields listed in Arg โ€œreference_fieldsโ€.

  3. โ€œmetricsโ€ โ€“ to contain the value of Arg โ€˜metricsโ€™

References: metrics.f1_binary, metrics.accuracy

Read more about catalog usage here.