You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Found an issue similar to #476, but for sklearn.neural_network.MLPClassifier objects.
Any model trained using an unsigned integer output type fails to be converted. See below a script to reproduce the issue:
The conversion fails for rettype = np.uint8, np.uint16 and np.uint32, but not for signed integer types or floating-point types.
A workaround can be to modify the model output type after training, before calling convert_sklearn: model.classes_ = model.classes_.astype( np.int32 )
Tested with Python 3.9, scikit-learn v1.0.1, skl2onnx v1.10.3
The text was updated successfully, but these errors were encountered:
Hi Team,
Found an issue similar to #476, but for sklearn.neural_network.MLPClassifier objects.
Any model trained using an unsigned integer output type fails to be converted. See below a script to reproduce the issue:
The conversion fails for rettype = np.uint8, np.uint16 and np.uint32, but not for signed integer types or floating-point types.
A workaround can be to modify the model output type after training, before calling convert_sklearn:
model.classes_ = model.classes_.astype( np.int32 )
Tested with Python 3.9, scikit-learn v1.0.1, skl2onnx v1.10.3
The text was updated successfully, but these errors were encountered: