Setting up the model schema
We use the Schema
class together with the from_df
method to set up a schema from the reference data.
In this case, we define the problem as 'BINARY_CLASSIFICATION' but other options like 'MULTICLASS_CLASSIFICATION' and 'REGRESSION' are possible.
More info about the Schema
class can be found in its API reference.
# Inspect schema from dataset and apply overrides
schema = nml_sdk.monitoring.Schema.from_df(
'BINARY_CLASSIFICATION',
reference_data,
target_column_name='work_home_actual',
ignore_column_names=('period'),
identifier_column_name='id'
)
Last updated