Packages

class ONNXModel extends Transformer with ComplexParamsWritable with ONNXModelParams with Wrappable with SynapseMLLogging

Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ONNXModel
  2. SynapseMLLogging
  3. Wrappable
  4. DotnetWrappable
  5. RWrappable
  6. PythonWrappable
  7. BaseWrappable
  8. ONNXModelParams
  9. HasFeedFetchDicts
  10. HasMiniBatcher
  11. ComplexParamsWritable
  12. MLWritable
  13. Transformer
  14. PipelineStage
  15. Logging
  16. Params
  17. Serializable
  18. Serializable
  19. Identifiable
  20. AnyRef
  21. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ONNXModel()
  2. new ONNXModel(uid: String)

Value Members

  1. val argMaxDict: StringStringMapParam
    Definition Classes
    ONNXModelParams
  2. final def clear(param: Param[_]): ONNXModel.this.type
    Definition Classes
    Params
  3. def copy(extra: ParamMap): ONNXModel
    Definition Classes
    ONNXModel → Transformer → PipelineStage → Params
  4. val deviceType: Param[String]
    Definition Classes
    ONNXModelParams
  5. def dotnetAdditionalMethods: String
    Definition Classes
    DotnetWrappable
  6. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  7. def explainParams(): String
    Definition Classes
    Params
  8. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  9. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  10. val feedDict: StringStringMapParam
    Definition Classes
    HasFeedFetchDicts
  11. val fetchDict: StringStringMapParam
    Definition Classes
    HasFeedFetchDicts
  12. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  13. def getArgMaxDict: Map[String, String]
    Definition Classes
    ONNXModelParams
  14. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  15. def getDeviceType: String
    Definition Classes
    ONNXModelParams
  16. def getFeedDict: Map[String, String]
    Definition Classes
    HasFeedFetchDicts
  17. def getFetchDict: Map[String, String]
    Definition Classes
    HasFeedFetchDicts
  18. def getMiniBatchSize: Int
    Definition Classes
    HasMiniBatcher
  19. def getMiniBatcher: MiniBatchBase

    Definition Classes
    HasMiniBatcher
  20. def getModelPayload: Array[Byte]
    Definition Classes
    ONNXModelParams
  21. def getOptimizationLevel: String
    Definition Classes
    ONNXModelParams
  22. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  23. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  24. def getParamInfo(p: Param[_]): ParamInfo[_]
    Definition Classes
    BaseWrappable
  25. def getSoftMaxDict: Map[String, String]
    Definition Classes
    ONNXModelParams
  26. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  27. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  28. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  29. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  30. def logClass(featureName: String): Unit
    Definition Classes
    SynapseMLLogging
  31. def logFit[T](f: ⇒ T, columns: Int): T
    Definition Classes
    SynapseMLLogging
  32. def logTransform[T](f: ⇒ T, columns: Int): T
    Definition Classes
    SynapseMLLogging
  33. def logVerb[T](verb: String, f: ⇒ T, columns: Option[Int] = None): T
    Definition Classes
    SynapseMLLogging
  34. def makeDotnetFile(conf: CodegenConfig): Unit
    Definition Classes
    DotnetWrappable
  35. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  36. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  37. val miniBatcher: TransformerParam

    Size of minibatches.

    Size of minibatches. Must be greater than 0; default is 10

    Definition Classes
    HasMiniBatcher
  38. def modelInput: Map[String, NodeInfo]
  39. def modelInputJava: Map[String, NodeInfo]
  40. def modelOutput: Map[String, NodeInfo]
  41. def modelOutputJava: Map[String, NodeInfo]
  42. val modelPayload: ByteArrayParam
    Definition Classes
    ONNXModelParams
  43. val optimizationLevel: Param[String]
    Definition Classes
    ONNXModelParams
  44. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  45. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  46. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  47. def rebroadcastModelPayload(spark: SparkSession): Broadcast[Array[Byte]]
  48. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  49. final def set[T](param: Param[T], value: T): ONNXModel.this.type
    Definition Classes
    Params
  50. def setArgMaxDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  51. def setArgMaxDict(value: HashMap[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  52. def setArgMaxDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  53. def setDeviceType(value: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  54. def setFeedDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  55. def setFeedDict(value: HashMap[String, String]): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  56. def setFeedDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  57. def setFetchDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  58. def setFetchDict(value: HashMap[String, String]): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  59. def setFetchDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  60. def setMiniBatchSize(n: Int): ONNXModel.this.type
    Definition Classes
    HasMiniBatcher
  61. def setMiniBatcher(value: MiniBatchBase): ONNXModel.this.type

    Definition Classes
    HasMiniBatcher
  62. def setModelLocation(path: String): ONNXModel.this.type
  63. def setModelPayload(value: Array[Byte]): ONNXModel.this.type
  64. def setOptimizationLevel(value: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  65. def setSoftMaxDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  66. def setSoftMaxDict(value: HashMap[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  67. def setSoftMaxDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  68. def sliceAtOutput(output: String): ONNXModel
  69. def sliceAtOutputs(outputs: Array[String]): ONNXModel
  70. val softMaxDict: StringStringMapParam
    Definition Classes
    ONNXModelParams
  71. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  72. def transform(dataset: Dataset[_]): DataFrame
    Definition Classes
    ONNXModel → Transformer
  73. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )
  74. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()
  75. def transformInner(dataset: Dataset[_], inputSchema: StructType): DataFrame
  76. def transformSchema(schema: StructType): StructType
    Definition Classes
    ONNXModel → PipelineStage
  77. val uid: String
    Definition Classes
    ONNXModelSynapseMLLogging → Identifiable
  78. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable