concrete.ml.pytest.utils.md

module concrete.ml.pytest.utils

Common functions or lists for test files, which can't be put in fixtures.

Global Variables

• regressor_models

• classifier_models

• classifiers

• regressors

function sanitize_test_and_train_datasets

sanitize_test_and_train_datasets(model, x, y)

Sanitize datasets depending on the model type.

Args:

• model: the model

• x: the first output of load_data, i.e., the inputs

• y: the second output of load_data, i.e., the labels

Returns: Tuple containing sanitized (model_params, x, y, x_train, y_train, x_test)