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