Examples

Below is a list of detailed example notebooks that go through the treatment effect estimation process for different ML frameworks using simulated data.

• ONNX Conversion: basic conversion tutorial for supported frameworks

• Check Conversion: post ONNX conversion test data generation and verification

• Sklearn: model training, conversion, data generation, and estimation

• Pytorch: neural network with categorical embeddings

• Tensorflow: SSDMobilenet conversion and estimation

• LightGBM: model training, conversion, and estimation