Packages

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

Linear Supertypes
BasicLogging, Wrappable, DotnetWrappable, 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. 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): Transformer
    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(): Unit
    Definition Classes
    BasicLogging
  31. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  32. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  33. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  34. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  35. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  36. def makeDotnetFile(conf: CodegenConfig): Unit
    Definition Classes
    DotnetWrappable
  37. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  38. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  39. val miniBatcher: TransformerParam

    Size of minibatches.

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

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

    Definition Classes
    HasMiniBatcher
  64. def setModelLocation(path: String): ONNXModel.this.type
  65. def setModelPayload(value: Array[Byte]): ONNXModel.this.type
  66. def setOptimizationLevel(value: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  67. def setSoftMaxDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  68. def setSoftMaxDict(value: HashMap[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  69. def setSoftMaxDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  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 transformSchema(schema: StructType): StructType
    Definition Classes
    ONNXModel → PipelineStage
  76. val uid: String
    Definition Classes
    ONNXModel → BasicLogging → Identifiable
  77. val ver: String
    Definition Classes
    BasicLogging
  78. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable