Infer types from the dataframe types
MainRo opened this issue · comments
- ebm2onnx version: 1.1.0
- Python version: 3.7
- Operating System: Linux
Description
Writing the type description for all features can be annoying. We could add a helper to infer the types automatically from a pandas dataframe:
onnx_model = ebm2onnx.to_onnx(
model,
dtype=ebm2onnx.get_types_from_pandas(df)
)