validmind.customer_churn

get_demo_test_config

defget_demo_test_config(test_suite=None):

Returns input configuration for the default documentation template assigned to this demo model

The default documentation template uses the following inputs:

We assign the following inputs depending on the input config expected by each test:

When a test expects a "dataset" we use the raw_dataset
When a tets expects "datasets" we use the train_dataset and test_dataset
When a test expects a "model" we use the model
When a test expects "model" and "dataset" we use the model and test_dataset
The only exception is ClassifierPerformance since that runs twice: once with the train_dataset (in sample) and once with the test_dataset (out of sample)

defload_data(full_dataset=False):

defpreprocess(df):