trait ONNXModelParams extends Params with HasMiniBatcher with HasFeedFetchDicts
Linear Supertypes
Known Subclasses
Ordering
- Alphabetic
- By Inheritance
Inherited
- ONNXModelParams
- HasFeedFetchDicts
- HasMiniBatcher
- Params
- Serializable
- Serializable
- Identifiable
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Abstract Value Members
Concrete Value Members
- val argMaxDict: StringStringMapParam
-
final
def
clear(param: Param[_]): ONNXModelParams.this.type
- Definition Classes
- Params
- val deviceType: Param[String]
-
def
explainParam(param: Param[_]): String
- Definition Classes
- Params
-
def
explainParams(): String
- Definition Classes
- Params
-
final
def
extractParamMap(): ParamMap
- Definition Classes
- Params
-
final
def
extractParamMap(extra: ParamMap): ParamMap
- Definition Classes
- Params
-
val
feedDict: StringStringMapParam
- Definition Classes
- HasFeedFetchDicts
-
val
fetchDict: StringStringMapParam
- Definition Classes
- HasFeedFetchDicts
-
final
def
get[T](param: Param[T]): Option[T]
- Definition Classes
- Params
- def getArgMaxDict: Map[String, String]
-
final
def
getDefault[T](param: Param[T]): Option[T]
- Definition Classes
- Params
- def getDeviceType: String
-
def
getFeedDict: Map[String, String]
- Definition Classes
- HasFeedFetchDicts
-
def
getFetchDict: Map[String, String]
- Definition Classes
- HasFeedFetchDicts
-
def
getMiniBatchSize: Int
- Definition Classes
- HasMiniBatcher
-
def
getMiniBatcher: MiniBatchBase
- Definition Classes
- HasMiniBatcher
- def getModelPayload: Array[Byte]
- def getOptimizationLevel: String
-
final
def
getOrDefault[T](param: Param[T]): T
- Definition Classes
- Params
-
def
getParam(paramName: String): Param[Any]
- Definition Classes
- Params
- def getSoftMaxDict: Map[String, String]
-
final
def
hasDefault[T](param: Param[T]): Boolean
- Definition Classes
- Params
-
def
hasParam(paramName: String): Boolean
- Definition Classes
- Params
-
final
def
isDefined(param: Param[_]): Boolean
- Definition Classes
- Params
-
final
def
isSet(param: Param[_]): Boolean
- Definition Classes
- Params
-
val
miniBatcher: TransformerParam
Size of minibatches.
Size of minibatches. Must be greater than 0; default is 10
- Definition Classes
- HasMiniBatcher
- val modelPayload: ByteArrayParam
- val optimizationLevel: Param[String]
-
lazy val
params: Array[Param[_]]
- Definition Classes
- Params
-
final
def
set[T](param: Param[T], value: T): ONNXModelParams.this.type
- Definition Classes
- Params
- def setArgMaxDict(k: String, v: String): ONNXModelParams.this.type
- def setArgMaxDict(value: HashMap[String, String]): ONNXModelParams.this.type
- def setArgMaxDict(value: Map[String, String]): ONNXModelParams.this.type
- def setDeviceType(value: String): ONNXModelParams.this.type
-
def
setFeedDict(k: String, v: String): ONNXModelParams.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFeedDict(value: HashMap[String, String]): ONNXModelParams.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFeedDict(value: Map[String, String]): ONNXModelParams.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFetchDict(k: String, v: String): ONNXModelParams.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFetchDict(value: HashMap[String, String]): ONNXModelParams.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFetchDict(value: Map[String, String]): ONNXModelParams.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setMiniBatchSize(n: Int): ONNXModelParams.this.type
- Definition Classes
- HasMiniBatcher
-
def
setMiniBatcher(value: MiniBatchBase): ONNXModelParams.this.type
- Definition Classes
- HasMiniBatcher
- def setOptimizationLevel(value: String): ONNXModelParams.this.type
- def setSoftMaxDict(k: String, v: String): ONNXModelParams.this.type
- def setSoftMaxDict(value: HashMap[String, String]): ONNXModelParams.this.type
- def setSoftMaxDict(value: Map[String, String]): ONNXModelParams.this.type
- val softMaxDict: StringStringMapParam
-
def
toString(): String
- Definition Classes
- Identifiable → AnyRef → Any