interpretml / ebm2onnx

A tool to convert EBM models to ONNX

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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)
)