task_type_check#
- task_type_check(model: BasicModel, dataset: Dataset) ModelType [source]#
Check task type (regression, binary, multiclass) according to model object and label column.
- Parameters
- modelBasicModel
Model object - used to check if it has predict_proba()
- datasettabular.Dataset
dataset - used to count the number of unique labels
- Returns
- ModelType
TaskType enum corresponding to the model and dataset