Packages

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

Linear Supertypes
SynapseMLLogging, 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. SynapseMLLogging
  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: 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 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: StringStringMapParam
    Definition Classes
    HasFeedFetchDicts
  10. val fetchDict: StringStringMapParam
    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 getParamInfo(p: Param[_]): ParamInfo[_]
    Definition Classes
    BaseWrappable
  24. def getSoftMaxDict: Map[String, String]
    Definition Classes
    ONNXModelParams
  25. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  26. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  27. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  28. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  29. def logClass(featureName: String): Unit
    Definition Classes
    SynapseMLLogging
  30. def logFit[T](f: ⇒ T, columns: Int): T
    Definition Classes
    SynapseMLLogging
  31. def logTransform[T](f: ⇒ T, columns: Int): T
    Definition Classes
    SynapseMLLogging
  32. def logVerb[T](verb: String, f: ⇒ T, columns: Option[Int] = None): T
    Definition Classes
    SynapseMLLogging
  33. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  34. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  35. val miniBatcher: TransformerParam

    Size of minibatches.

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

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

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