concrete.ml.pytest.utils.md

Previousconcrete.ml.pytest.torch_models.md Nextconcrete.ml.quantization.base_quantized_op.md

Last updated 2 years ago

Was this helpful?

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)