onnxmltools icon indicating copy to clipboard operation
onnxmltools copied to clipboard

Published 20 hours ago • verified publisher icon onnx

sparkml utils.py getTensorTypeFromSpark for batch inputs; buildInputDictSimple excludes StringType()

Open mattoh91 opened this issue 1 year ago • 1 comments

  1. getTensorTypeFromSpark() returns tensor types of shape [1,1]: This should be updated to [None,1] to accommodate batch data input.
  2. buildInputDictSimple() only accounts for 'StringType' string dataType: This should also include for 'StringType()'

mattoh91 avatar Mar 20 '23 07:03 mattoh91

Submitted PR to fix issues mentioned above.

mattoh91 avatar Mar 20 '23 07:03 mattoh91