Giters

interpretml / ebm2onnx

A tool to convert EBM models to ONNX

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The output value of a classification is an index instead of the class

MainRo opened this issue · comments

commented
  • ebm2onnx version: v3.1.0
  • Python version: 3.9
  • Operating System: Linux

Description

When converting a classifier that outputs strings for each class, then the prediction output is bad: It corresponds to the class index instead of the class name. This is inconsistent with the output of ebm.predict.