Skip to content

Commit 2f96f95

Browse files
authored
Update index.rst (#532)
* Update index.rst change 1 to None so the inference can run Signed-off-by: xiaowuhu <[email protected]> * Update index.rst add comment Signed-off-by: xiaowuhu <[email protected]>
1 parent 158f2a2 commit 2f96f95

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/index.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ ONNX and finally uses *onnxruntime* to predict.
5959
from onnxmltools import convert_sklearn
6060
from onnxmltools.utils import save_model
6161
from onnxconverter_common.data_types import FloatTensorType
62-
initial_type = [('float_input', FloatTensorType([1, 4]))]
62+
initial_type = [('float_input', FloatTensorType([None, 4]))] # using 'None' to support multiple input samples for pred
6363
onx = convert_sklearn(clr, initial_types=initial_type)
6464
save_model(onx, "rf_iris.onnx")
6565

0 commit comments

Comments
 (0)