Add shape to FloatTensor probabilities (#600)

When zipmap is set to false, onnx checker complains about
'probabilities' tensor that do not have a shape. This commit adds
correct shape for this case.

Signed-off-by: Frederic DALENCON <[email protected]>

Signed-off-by: Frederic DALENCON <[email protected]>

Author: frdasah

onnxmltools/convert/lightgbm/_parse.py

@@ -148,7 +148,7 @@ def _parse_sklearn_classifier(scope, model, inputs, zipmap=True):
             SequenceType(DictionaryType(label_type, FloatTensorType())))
     else:
         output_probability = scope.declare_local_variable(
-            'probabilities', FloatTensorType())
+            'probabilities', FloatTensorType(shape=[None, len(classes)]))
     this_operator.outputs.append(output_label)
     this_operator.outputs.append(output_probability)
     return this_operator.outputs