evalml.data_checks.ClassImbalanceDataCheck

class evalml.data_checks.ClassImbalanceDataCheck(threshold=0.1, num_cv_folds=3)[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 target labels are imbalanced, or if the number of values for each target

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

Class Inheritance

Inheritance diagram of ClassImbalanceDataCheck