task_inference#

Utils module containing functionalities to infer the task type and possible label classes.

Functions

infer_task_type_by_labels(labels)

Infer task type from given dataset/labels/model_classes.

infer_classes_from_model(model)

Get classes_ attribute from model object if exists.

get_all_labels(model, train_dataset[, ...])

Aggregate labels from all available data: labels on datasets, y_pred, and model predicitions.

infer_task_type_by_class_number(num_classes)

Infer task type of binary or multiclass.