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.


  • 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)

Last updated