问在字符串输入上使用onnx运行时进行预测

EN

'''
#convert pipeline into onnx
model_onnx = convert_sklearn(pipeline, "tfidf",
                             initial_types=[("str_input", StringTensorType([1, 2000]))])

with open("pipeline_emails.onnx", "wb") as f:
     f.write(onx.SerializeToString())

#make predictions on test data
sess = rt.InferenceSession("pipeline_emails.onnx")
pred_onx = sess.run(None, {"str_input": test_df.as_matrix()})[0]
print("predict", pred_onx[0])
print("predict_proba", pred_onx[1])
'''

RuntimeError                              Traceback (most recent call last)
<ipython-input-118-5db056b989a8> in <module>()
      2 sess = rt.InferenceSession("pipeline_emails.onnx")
      3 inputs = {'str_input': test_df.as_matrix()}
----> 4 pred_onx = sess.run(None, {"str_input": test_df.as_matrix()})[0]
      5 print("predict", pred_onx[0])
      6 print("predict_proba", pred_onx[1])

~\AppData\Local\Continuum\anaconda3\lib\site-packages\onnxruntime\capi\session.py in run(self, output_names, input_feed, run_options)
     70         if not output_names:
     71             output_names = [output.name for output in self._outputs_meta]
---> 72         return self._sess.run(output_names, input_feed, run_options)
     73 
     74     def end_profiling(self):

RuntimeError: Method run failed due to: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Missing required input: float_input