evalml.data_checks.ClassImbalanceDataCheck

Inheritance diagram of ClassImbalanceDataCheck
class evalml.data_checks.ClassImbalanceDataCheck(threshold=0.1)[source]

Checks if any target labels are imbalanced beyond a threshold. Use for classification problems

name = 'ClassImbalanceDataCheck'

Instance attributes

Methods:

__init__

Check if any of the features are likely to be ID columns.

validate

Checks if any target labels are imbalanced beyond a threshold for binary and multiclass problems Ignores nan values in target labels if they appear