DashAI.back.models.BaseModel

class BaseModel[source]

Abstract class of all machine learning models.

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

__init__()

Methods

__init__()

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.

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