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. val argMaxDict: MapParam[String, String]
    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 explainParam(param: Param[_]): String
    Definition Classes
    Params
  6. def explainParams(): String
    Definition Classes
    Params
  7. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  8. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  9. val feedDict: MapParam[String, String]
    Definition Classes
    HasFeedFetchDicts
  10. val fetchDict: MapParam[String, String]
    Definition Classes
    HasFeedFetchDicts
  11. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  12. def getArgMaxDict: Map[String, String]
    Definition Classes
    ONNXModelParams
  13. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  14. def getDeviceType: String
    Definition Classes
    ONNXModelParams
  15. def getFeedDict: Map[String, String]
    Definition Classes
    HasFeedFetchDicts
  16. def getFetchDict: Map[String, String]
    Definition Classes
    HasFeedFetchDicts
  17. def getMiniBatchSize: Int
    Definition Classes
    HasMiniBatcher
  18. def getMiniBatcher: MiniBatchBase

    Definition Classes
    HasMiniBatcher
  19. def getModelPayload: Array[Byte]
    Definition Classes
    ONNXModelParams
  20. def getOptimizationLevel: String
    Definition Classes
    ONNXModelParams
  21. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  22. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  23. def getSoftMaxDict: Map[String, String]
    Definition Classes
    ONNXModelParams
  24. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  25. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  26. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  27. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  28. def logClass(): Unit
    Definition Classes
    BasicLogging
  29. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  30. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  31. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  32. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  33. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  34. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  35. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  36. val miniBatcher: TransformerParam

    Size of minibatches.

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

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

    Definition Classes
    HasMiniBatcher
  58. def setModelLocation(path: String): ONNXModel.this.type
  59. def setModelPayload(value: Array[Byte]): ONNXModel.this.type
  60. def setOptimizationLevel(value: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  61. def setSoftMaxDict(k: String, v: String): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  62. def setSoftMaxDict(value: Map[String, String]): ONNXModel.this.type
    Definition Classes
    ONNXModelParams
  63. val softMaxDict: MapParam[String, String]
    Definition Classes
    ONNXModelParams
  64. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  65. def transform(dataset: Dataset[_]): DataFrame
    Definition Classes
    ONNXModel → Transformer
  66. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )
  67. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()
  68. def transformSchema(schema: StructType): StructType
    Definition Classes
    ONNXModel → PipelineStage
  69. val uid: String
    Definition Classes
    ONNXModel → BasicLogging → Identifiable
  70. val ver: String
    Definition Classes
    BasicLogging
  71. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable