Synapseml
0.10.0
|
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. |