ML - Fixed script formatting for prediction (#11767)

* Fixed script formatting for prediction
This commit is contained in:
Leila Lali
2020-08-12 13:36:39 -07:00
committed by GitHub
parent 094ee7c50c
commit e40a81e8e1
5 changed files with 137 additions and 129 deletions

View File

@@ -64,13 +64,13 @@ export class ModelPythonClient {
'float', 'uint8', 'int8', 'uint16', 'int16', 'int32', 'int64', 'string', 'bool', 'double',
'uint32', 'uint64', 'complex64', 'complex128', 'bfloat16']`,
`type_map = {
onnx.TensorProto.DataType.FLOAT: 'real',
onnx.TensorProto.DataType.UINT8: 'tinyint',
onnx.TensorProto.DataType.INT16: 'smallint',
onnx.TensorProto.DataType.INT32: 'int',
onnx.TensorProto.DataType.INT64: 'bigint',
onnx.TensorProto.DataType.STRING: 'varchar(MAX)',
onnx.TensorProto.DataType.DOUBLE: 'float'}`,
onnx.TensorProto.DataType.FLOAT: 'REAL',
onnx.TensorProto.DataType.UINT8: 'TINYINT',
onnx.TensorProto.DataType.INT16: 'SMALLINT',
onnx.TensorProto.DataType.INT32: 'INT',
onnx.TensorProto.DataType.INT64: 'BIGINT',
onnx.TensorProto.DataType.STRING: 'VARCHAR(MAX)',
onnx.TensorProto.DataType.DOUBLE: 'FLOAT'}`,
`parameters = {
"inputs": [],
"outputs": []