Check input features of Model match Schema

NVIDIA-Merlin / Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

https://nvidia-merlin.github.io/Transformers4Rec/main

Apache License 2.0

1.08k stars 142 forks source link

Check input features of Model match Schema #713

Open oliverholworthy opened 1 year ago

oliverholworthy commented 1 year ago

Goals :soccer:

Ensure that the model is called with features that match the Model.input_schema. This ensures that we trace the model with the correct inputs when creating a model to serve in Triton.

Implementation Details :construction:

Validate that all columns are processed in TabularFeatures
- Raises exception if any columns are not used
Validate that the inputs match the model input_schema in the foward method of the model
- Raises exception if these don't match

NVIDIA-Merlin / Transformers4Rec

Check input features of Model match Schema #713

Goals :soccer:

Implementation Details :construction:

Testing Details :mag: