Customizing your Model Schema

There can be occasions when the fdl.ModelSchema object generated by fdl.Model.from_data infers a column's data type differently than the type intended by the model developer. In these cases you can modify the ModelSchema columns as needed prior to creating the model in Fiddler.

Let's walk through an example of how to do this.


Suppose you've loaded in a dataset as a pandas DataFrame.

import pandas as pd

df = pd.read_csv('example_dataset.csv')

Below is an example of what is displayed upon inspection.


You then create a fdl.Model object by inferring the column schema details from this DataFrame.

model = fdl.Model.from_data(
  name='my_model',
  project_id=PROJECT_ID,
  source=df
)

Below is an example of what is displayed upon inspection of model.schema.

Upon inspection you may notice that a few things are off:

  1. The value range of output_column is set to [0.01, 0.99], when it should really be [0.0, 1.0].

  2. There are no possible values set for feature_3.

  3. The data type of feature_3 is set to fdl.DataType.STRING, when it should really be fdl.DataType.CATEGORY.

What's the downside of not making sure that ranges and categories are reviewed and properly set? A production traffic event that encodes a number outside of the specified value range or a category value that is not in the set of valid category values will further down the line be flagged as a so-called Data Integrity violation. Depending on the alerting config, this may result in an alert. It's also worth noting however that an event which has a violation in its columns is still processed, and metrics that can be generated are still generated.

The below examples demonstrate how to address each of the issues noted:

Modifying a Column’s Value Range

Let's say we want to modify the range of output_column in the above fdl.Model object to be [0.0, 1.0].

You can do this by setting the min and max of the output_column column.

model.schema['output_column'].min = 0.0
model.schema['output_column'].max = 1.0

Modifying a Column’s Possible Values

Let's say we want to modify the possible values of feature_3 to be ['Yes', 'No'].

You can do this by setting the categories of the feature_3 column.

model.schema['feature_3'].categories = ['Yes', 'No']

Modifying a Column’s Data Type

Let's say we want to modify the data type of feature_3 to be fdl.DataType.CATEGORY.

You can do this by setting the data_type of the feature_3 column.

model.schema['feature_3'].data_type = fdl.DataType.CATEGORY

🚧 Note when modifying a column's data type to Category:

It is also required to set the column's possible_values to the list of unique possible values for that column. This is done automatically when a String column is inferred as a Category.

model.schema['feature_3'].data_type = fdl.DataType.CATEGORY model.schema['feature_3'].possible_values = ['Yes', 'No']

Last updated

© 2024 Fiddler AI