label

Classes:

MultilabelIssueManager(datalab, **_)

Manages label issues in Datalab for multilabel tasks.

class cleanlab.datalab.internal.issue_manager.multilabel.label.MultilabelIssueManager(datalab, **_)

Bases: IssueManager

Manages label issues in Datalab for multilabel tasks.

Parameters:

datalab (Datalab) – A Datalab instance.

Attributes:

description	Short text that summarizes the type of issues handled by this IssueManager.
issue_name	Returns a key that is used to store issue summary results about the assigned Lab.
verbosity_levels	A dictionary of verbosity levels and their corresponding dictionaries of report items to print.
issue_score_key	Returns a key that is used to store issue score results about the assigned Lab.
info
issues
summary

Methods:

find_issues(pred_probs, **kwargs)	Find label issues in a multilabel dataset.
make_summary(score)	Construct a summary dataframe.
report(issues, summary, info[, ...])	Compose a report of the issues found by this IssueManager.
collect_info(given_labels, predicted_labels)	Collects data for the info attribute of the Datalab.

description: ClassVar[str]

Short text that summarizes the type of issues handled by this IssueManager.

issue_name: ClassVar[str] = 'label'

Returns a key that is used to store issue summary results about the assigned Lab.

verbosity_levels: ClassVar[Dict[int, List[str]]]

A dictionary of verbosity levels and their corresponding dictionaries of report items to print.

Example

>>> verbosity_levels = {
...     0: [],
...     1: ["some_info_key"],
...     2: ["additional_info_key"],
... }

find_issues(pred_probs, **kwargs)

Find label issues in a multilabel dataset.

Parameters:

pred_probs (ndarray[Any, dtype[TypeVar(ScalarType, bound= generic, covariant=True)]]) – The predicted probabilities for each example.

Return type:

None

issue_score_key: ClassVar[str] = 'label_score'

Returns a key that is used to store issue score results about the assigned Lab.

classmethod make_summary(score)

Construct a summary dataframe.

Parameters:

score (float) – The overall score for this issue.

Return type:

DataFrame

Returns:

summary – A summary dataframe.

classmethod report(issues, summary, info, num_examples=5, verbosity=0, include_description=False, info_to_omit=None)

Compose a report of the issues found by this IssueManager.

Parameters:

• issues (DataFrame) –

  An issues dataframe.

  Example

  >>> import pandas as pd
  >>> issues = pd.DataFrame(
  ...     {
  ...         "is_X_issue": [True, False, True],
  ...         "X_score": [0.2, 0.9, 0.4],
  ...     },
  ... )
  

• summary (DataFrame) –

  The summary dataframe.

  Example

  >>> summary = pd.DataFrame(
  ...     {
  ...         "issue_type": ["X"],
  ...         "score": [0.5],
  ...     },
  ... )
  

• info (Dict[str, Any]) –

  The info dict.

  Example

  >>> info = {
  ...     "A": "val_A",
  ...     "B": ["val_B1", "val_B2"],
  ... }
  

• num_examples (int) – The number of examples to print.

• verbosity (int) – The verbosity level of the report.

• include_description (bool) – Whether to include a description of the issue in the report.

Return type:

str

Returns:

report_str – A string containing the report.

info: Dict[str, Any]

issues: DataFrame

summary: DataFrame

collect_info(given_labels, predicted_labels)

Collects data for the info attribute of the Datalab. :rtype: dict

Note

This method is called by find_issues() after find_issues() has set the issues and summary dataframes as instance attributes.