|
Synapseml
1.0.4
|
ONNXModel implements ONNXModel More...


Public Member Functions | |
| ONNXModel () | |
| Creates a ONNXModel without any parameters. More... | |
| ONNXModel (string uid) | |
| Creates a ONNXModel with a UID that is used to give the ONNXModel a unique ID. More... | |
| ONNXModel | SetArgMaxDict (Dictionary< string, string > value) |
| Sets value for argMaxDict More... | |
| ONNXModel | SetDeviceType (string value) |
| Sets value for deviceType More... | |
| ONNXModel | SetFeedDict (Dictionary< string, string > value) |
| Sets value for feedDict More... | |
| ONNXModel | SetFetchDict (Dictionary< string, string > value) |
| Sets value for fetchDict More... | |
| ONNXModel | SetMiniBatcher (JavaTransformer value) |
| Sets value for miniBatcher More... | |
| ONNXModel | SetModelPayload (object value) |
| Sets value for modelPayload More... | |
| ONNXModel | SetOptimizationLevel (string value) |
| Sets value for optimizationLevel More... | |
| ONNXModel | SetSoftMaxDict (Dictionary< string, string > value) |
| Sets value for softMaxDict More... | |
| Dictionary< string, string > | GetArgMaxDict () |
| Gets argMaxDict value More... | |
| string | GetDeviceType () |
| Gets deviceType value More... | |
| Dictionary< string, string > | GetFeedDict () |
| Gets feedDict value More... | |
| Dictionary< string, string > | GetFetchDict () |
| Gets fetchDict value More... | |
| JavaTransformer | GetMiniBatcher () |
| Gets miniBatcher value More... | |
| object | GetModelPayload () |
| Gets modelPayload value More... | |
| string | GetOptimizationLevel () |
| Gets optimizationLevel value More... | |
| Dictionary< string, string > | GetSoftMaxDict () |
| Gets softMaxDict value More... | |
| void | Save (string path) |
| Saves the object so that it can be loaded later using Load. Note that these objects can be shared with Scala by Loading or Saving in Scala. More... | |
| JavaMLWriter | Write () |
| |
| JavaMLReader< ONNXModel > | Read () |
| Get the corresponding JavaMLReader instance. More... | |
Static Public Member Functions | |
| static ONNXModel | Load (string path) |
| Loads the ONNXModel that was previously saved using Save(string). More... | |
|
inline |
Creates a ONNXModel without any parameters.
|
inline |
|
inline |
Gets argMaxDict value
| string Synapse.ML.Onnx.ONNXModel.GetDeviceType | ( | ) |
Gets deviceType value
|
inline |
Gets feedDict value
|
inline |
Gets fetchDict value
|
inline |
Gets miniBatcher value
| object Synapse.ML.Onnx.ONNXModel.GetModelPayload | ( | ) |
Gets modelPayload value
| string Synapse.ML.Onnx.ONNXModel.GetOptimizationLevel | ( | ) |
Gets optimizationLevel value
|
inline |
Gets softMaxDict value
Map("rawPrediction" -> "probability") to obtain the probability outputs.
|
static |
Loads the ONNXModel that was previously saved using Save(string).
| path | The path the previous ONNXModel was saved to |
| JavaMLReader<ONNXModel> Synapse.ML.Onnx.ONNXModel.Read | ( | ) |
Get the corresponding JavaMLReader instance.
| void Synapse.ML.Onnx.ONNXModel.Save | ( | string | path | ) |
Saves the object so that it can be loaded later using Load. Note that these objects can be shared with Scala by Loading or Saving in Scala.
| path | The path to save the object to |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetArgMaxDict | ( | Dictionary< string, string > | value | ) |
Sets value for argMaxDict
| value | A map between output dataframe columns, where the value column will be computed from taking the argmax of the key column. This can be used to convert probability output to predicted label. |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetDeviceType | ( | string | value | ) |
Sets value for deviceType
| value | Specify a device type the model inference runs on. Supported types are: CPU or CUDA.If not specified, auto detection will be used. |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetFeedDict | ( | Dictionary< string, string > | value | ) |
Sets value for feedDict
| value | Provide a map from CNTK/ONNX model input variable names (keys) to column names of the input dataframe (values) |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetFetchDict | ( | Dictionary< string, string > | value | ) |
Sets value for fetchDict
| value | Provide a map from column names of the output dataframe (keys) to CNTK/ONNX model output variable names (values) |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetMiniBatcher | ( | JavaTransformer | value | ) |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetModelPayload | ( | object | value | ) |
Sets value for modelPayload
| value | Array of bytes containing the serialized ONNX model. |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetOptimizationLevel | ( | string | value | ) |
Sets value for optimizationLevel
| value | Specify the optimization level for the ONNX graph optimizations. Details at https://onnxruntime.ai/docs/resources/graph-optimizations.html#graph-optimization-levels. Supported values are: NO_OPT; BASIC_OPT; EXTENDED_OPT; ALL_OPT. Default: ALL_OPT. |
| ONNXModel Synapse.ML.Onnx.ONNXModel.SetSoftMaxDict | ( | Dictionary< string, string > | value | ) |
Sets value for softMaxDict
| value | A map between output dataframe columns, where the value column will be computed from taking the softmax of the key column. If the 'rawPrediction' column contains logits outputs, then one can set softMaxDict to Map("rawPrediction" -> "probability") to obtain the probability outputs. |
1.8.17