DashAI.back.models.BaseModel

class BaseModel[source]

Abstract class of all machine learning models.

All models must extend this class and implement save, load, train and predict methods.

__init__()

Methods

__init__()

calculate_metrics([split, level, log_index, ...])

Calculate and save metrics for a given data split and level.

get_schema()

Generates the component related Json Schema.

load(filename)

Restores an instance of a model.

prepare_dataset(dataset[, is_fit])

Hook for model-specific preprocessing of input features.

prepare_output(dataset[, is_fit])

Hook for model-specific preprocessing of output targets.

save(filename)

Store an instance of a model.

train(x_train, y_train[, x_validation, ...])

Train the model with the provided data.

validate_and_transform(raw_data)

It takes the data given by the user to initialize the model and returns it with all the objects that the model needs to work.

Attributes

COLOR

TYPE