Packages

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

Linear Supertypes
BasicLogging, Wrappable, RWrappable, PythonWrappable, BaseWrappable, ONNXModelParams, HasFeedFetchDicts, HasMiniBatcher, ComplexParamsWritable, MLWritable, Transformer, PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ONNXModel
  2. BasicLogging
  3. Wrappable
  4. RWrappable
  5. PythonWrappable
  6. BaseWrappable
  7. ONNXModelParams
  8. HasFeedFetchDicts
  9. HasMiniBatcher
  10. ComplexParamsWritable
  11. MLWritable
  12. Transformer
  13. PipelineStage
  14. Logging
  15. Params
  16. Serializable
  17. Serializable
  18. Identifiable
  19. AnyRef
  20. 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. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  5. val argMaxDict: MapParam[String, String]
    Definition Classes
    ONNXModelParams
  6. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  7. lazy val classNameHelper: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  8. final def clear(param: Param[_]): ONNXModel.this.type
    Definition Classes
    Params
  9. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  10. def companionModelClassName: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  11. def copy(extra: ParamMap): Transformer
    Definition Classes
    ONNXModel → Transformer → PipelineStage → Params
  12. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  13. lazy val copyrightLines: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  14. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  15. val deviceType: Param[String]
    Definition Classes
    ONNXModelParams
  16. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  17. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  18. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  19. def explainParams(): String
    Definition Classes
    Params
  20. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  21. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  22. val feedDict: MapParam[String, String]
    Definition Classes
    HasFeedFetchDicts
  23. val fetchDict: MapParam[String, String]
    Definition Classes
    HasFeedFetchDicts
  24. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  25. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  26. def getArgMaxDict: Map[String, String]
    Definition Classes
    ONNXModelParams
  27. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  28. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  29. def getDeviceType: String
    Definition Classes
    ONNXModelParams
  30. def getFeedDict: Map[String, String]
    Definition Classes
    HasFeedFetchDicts
  31. def getFetchDict: Map[String, String]
    Definition Classes
    HasFeedFetchDicts
  32. def getMiniBatchSize: Int
    Definition Classes
    HasMiniBatcher
  33. def getMiniBatcher: MiniBatchBase

    Definition Classes
    HasMiniBatcher
  34. def getModelPayload: Array[Byte]
    Definition Classes
    ONNXModelParams
  35. def getOptimizationLevel: String
    Definition Classes
    ONNXModelParams
  36. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  37. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  38. def getSoftMaxDict: Map[String, String]
    Definition Classes
    ONNXModelParams
  39. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  40. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  41. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  42. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  43. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  44. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  45. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  46. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  47. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  48. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  49. def logBase(methodName: String): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  50. def logClass(): Unit
    Definition Classes
    BasicLogging
  51. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  52. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  53. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  54. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  55. def logErrorBase(methodName: String, e: Exception): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  56. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  57. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  58. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  59. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  60. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  61. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  62. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  63. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  64. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  65. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  66. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  67. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  68. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  69. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  70. val miniBatcher: TransformerParam

    Size of minibatches.

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

    Definition Classes
    HasMiniBatcher
  71. def modelInput: Map[String, NodeInfo]
  72. def modelInputJava: Map[String, NodeInfo]
  73. def modelOutput: Map[String, NodeInfo]
  74. def modelOutputJava: Map[String, NodeInfo]
  75. val modelPayload: ByteArrayParam
    Definition Classes
    ONNXModelParams
  76. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  77. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  78. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  79. val optimizationLevel: Param[String]
    Definition Classes
    ONNXModelParams
  80. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  81. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  82. lazy val pyClassDoc: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  83. lazy val pyClassName: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  84. def pyExtraEstimatorImports: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  85. def pyExtraEstimatorMethods: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  86. lazy val pyInheritedClasses: Seq[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  87. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  88. lazy val pyInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    ONNXModel → PythonWrappable
  89. lazy val pyObjectBaseClass: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  90. def pyParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  91. def pyParamDefault[T](p: Param[T]): Option[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  92. def pyParamGetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  93. def pyParamSetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  94. def pyParamsArgs: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  95. def pyParamsDefaults: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  96. lazy val pyParamsDefinitions: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  97. def pyParamsGetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  98. def pyParamsSetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  99. def pythonClass(): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  100. def rClass(): String
    Attributes
    protected
    Definition Classes
    RWrappable
  101. def rDocString: String
    Attributes
    protected
    Definition Classes
    RWrappable
  102. def rExtraBodyLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  103. def rExtraInitLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  104. lazy val rFuncName: String
    Attributes
    protected
    Definition Classes
    RWrappable
  105. lazy val rInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    RWrappable
  106. def rParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    RWrappable
  107. def rParamsArgs: String
    Attributes
    protected
    Definition Classes
    RWrappable
  108. def rSetterLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  109. def rebroadcastModelPayload(spark: SparkSession): Broadcast[Array[Byte]]
  110. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  111. final def set(paramPair: ParamPair[_]): ONNXModel.this.type
    Attributes
    protected
    Definition Classes
    Params
  112. final def set(param: String, value: Any): ONNXModel.this.type
    Attributes
    protected
    Definition Classes
    Params
  113. final def set[T](param: Param[T], value: T): ONNXModel.this.type
    Definition Classes
    Params
  114. def setArgMaxDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  115. def setArgMaxDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  116. final def setDefault(paramPairs: ParamPair[_]*): ONNXModel.this.type
    Attributes
    protected
    Definition Classes
    Params
  117. final def setDefault[T](param: Param[T], value: T): ONNXModel.this.type
    Attributes
    protected
    Definition Classes
    Params
  118. def setDeviceType(value: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  119. def setFeedDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  120. def setFeedDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  121. def setFetchDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  122. def setFetchDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    HasFeedFetchDicts
  123. def setMiniBatchSize(n: Int): ONNXModel.this.type
    Definition Classes
    HasMiniBatcher
  124. def setMiniBatcher(value: MiniBatchBase): ONNXModel.this.type

    Definition Classes
    HasMiniBatcher
  125. def setModelLocation(path: String): ONNXModel.this.type
  126. def setModelPayload(value: Array[Byte]): ONNXModel.this.type
  127. def setOptimizationLevel(value: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  128. def setSoftMaxDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  129. def setSoftMaxDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  130. val softMaxDict: MapParam[String, String]
    Definition Classes
    ONNXModelParams
  131. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  132. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  133. def transform(dataset: Dataset[_]): DataFrame
    Definition Classes
    ONNXModel → Transformer
  134. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )
  135. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()
  136. def transformSchema(schema: StructType): StructType
    Definition Classes
    ONNXModel → PipelineStage
  137. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  138. val uid: String
    Definition Classes
    ONNXModel → BasicLogging → Identifiable
  139. val ver: String
    Definition Classes
    BasicLogging
  140. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  141. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  142. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  143. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable

Inherited from BasicLogging

Inherited from Wrappable

Inherited from RWrappable

Inherited from PythonWrappable

Inherited from BaseWrappable

Inherited from ONNXModelParams

Inherited from HasFeedFetchDicts

Inherited from HasMiniBatcher

Inherited from ComplexParamsWritable

Inherited from MLWritable

Inherited from Transformer

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

getParam

param

setParam

Ungrouped