class CNTKModel extends Model[CNTKModel] with ComplexParamsWritable with HasMiniBatcher with HasFeedFetchDicts with Wrappable with BasicLogging
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- CNTKModel
- BasicLogging
- Wrappable
- RWrappable
- PythonWrappable
- BaseWrappable
- HasFeedFetchDicts
- HasMiniBatcher
- ComplexParamsWritable
- MLWritable
- Model
- Transformer
- PipelineStage
- Logging
- Params
- Serializable
- Serializable
- Identifiable
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Value Members
- val batchInput: BooleanParam
-
final
def
clear(param: Param[_]): CNTKModel.this.type
- Definition Classes
- Params
- val convertOutputToDenseVector: BooleanParam
-
def
copy(extra: ParamMap): CNTKModel.this.type
- Definition Classes
- CNTKModel → Model → Transformer → PipelineStage → Params
-
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: MapParam[String, String]
- Definition Classes
- HasFeedFetchDicts
-
val
fetchDict: MapParam[String, String]
- Definition Classes
- HasFeedFetchDicts
-
final
def
get[T](param: Param[T]): Option[T]
- Definition Classes
- Params
- def getBatchInput: Boolean
- def getConvertOutputToDenseVector: Boolean
-
final
def
getDefault[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
def
getFeedDict: Map[String, String]
- Definition Classes
- HasFeedFetchDicts
-
def
getFetchDict: Map[String, String]
- Definition Classes
- HasFeedFetchDicts
- def getInputCol: String
- def getInputNode: String
- def getInputNodeIndex: Int
-
def
getInputShapes: List[Array[Int]]
Returns the dimensions of the required input
-
def
getMiniBatchSize: Int
- Definition Classes
- HasMiniBatcher
-
def
getMiniBatcher: MiniBatchBase
- Definition Classes
- HasMiniBatcher
- def getModel: SerializableFunction
-
final
def
getOrDefault[T](param: Param[T]): T
- Definition Classes
- Params
- def getOutputCol: String
- def getOutputNode: String
- def getOutputNodeIndex: Int
-
def
getParam(paramName: String): Param[Any]
- Definition Classes
- Params
-
final
def
hasDefault[T](param: Param[T]): Boolean
- Definition Classes
- Params
-
def
hasParam(paramName: String): Boolean
- Definition Classes
- Params
-
def
hasParent: Boolean
- Definition Classes
- Model
-
final
def
isDefined(param: Param[_]): Boolean
- Definition Classes
- Params
-
final
def
isSet(param: Param[_]): Boolean
- Definition Classes
- Params
-
def
logClass(): Unit
- Definition Classes
- BasicLogging
-
def
logFit[T](f: ⇒ T): T
- Definition Classes
- BasicLogging
-
def
logPredict[T](f: ⇒ T): T
- Definition Classes
- BasicLogging
-
def
logTrain[T](f: ⇒ T): T
- Definition Classes
- BasicLogging
-
def
logTransform[T](f: ⇒ T): T
- Definition Classes
- BasicLogging
-
def
logVerb[T](verb: String, f: ⇒ T): T
- Definition Classes
- BasicLogging
-
def
makePyFile(conf: CodegenConfig): Unit
- Definition Classes
- PythonWrappable
-
def
makeRFile(conf: CodegenConfig): Unit
- Definition Classes
- RWrappable
-
val
miniBatcher: TransformerParam
Size of minibatches.
Size of minibatches. Must be greater than 0; default is 10
- Definition Classes
- HasMiniBatcher
-
val
model: CNTKFunctionParam
Array of bytes containing the serialized CNTK
Function
-
lazy val
params: Array[Param[_]]
- Definition Classes
- Params
-
var
parent: Estimator[CNTKModel]
- Definition Classes
- Model
-
def
pyAdditionalMethods: String
- Definition Classes
- PythonWrappable
-
def
pyInitFunc(): String
- Definition Classes
- PythonWrappable
- def rebroadcastCNTKModel(spark: SparkSession): Unit
-
def
save(path: String): Unit
- Definition Classes
- MLWritable
- Annotations
- @Since( "1.6.0" ) @throws( ... )
-
final
def
set[T](param: Param[T], value: T): CNTKModel.this.type
- Definition Classes
- Params
- def setBatchInput(v: Boolean): CNTKModel.this.type
- def setConvertOutputToDenseVector(v: Boolean): CNTKModel.this.type
-
def
setFeedDict(k: String, v: String): CNTKModel.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFeedDict(value: Map[String, String]): CNTKModel.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFetchDict(k: String, v: String): CNTKModel.this.type
- Definition Classes
- HasFeedFetchDicts
-
def
setFetchDict(value: Map[String, String]): CNTKModel.this.type
- Definition Classes
- HasFeedFetchDicts
- def setInputCol(value: String): CNTKModel.this.type
- def setInputNode(value: String): CNTKModel.this.type
- def setInputNodeIndex(value: Int): CNTKModel.this.type
-
def
setMiniBatchSize(n: Int): CNTKModel.this.type
- Definition Classes
- HasMiniBatcher
-
def
setMiniBatcher(value: MiniBatchBase): CNTKModel.this.type
- Definition Classes
- HasMiniBatcher
- def setModel(value: SerializableFunction): CNTKModel.this.type
- def setModelLocation(path: String): CNTKModel.this.type
- def setOutputCol(value: String): CNTKModel.this.type
- def setOutputNode(value: String): CNTKModel.this.type
- def setOutputNodeIndex(value: Int): CNTKModel.this.type
-
def
setParent(parent: Estimator[CNTKModel]): CNTKModel
- Definition Classes
- Model
-
def
toString(): String
- Definition Classes
- Identifiable → AnyRef → Any
-
def
transform(dataset: Dataset[_]): DataFrame
Evaluate the model
Evaluate the model
- dataset
the dataset to featurize
- returns
featurized dataset
- Definition Classes
- CNTKModel → Transformer
-
def
transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame
- Definition Classes
- Transformer
- Annotations
- @Since( "2.0.0" )
-
def
transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame
- Definition Classes
- Transformer
- Annotations
- @Since( "2.0.0" ) @varargs()
-
def
transformSchema(schema: StructType): StructType
- Definition Classes
- CNTKModel → PipelineStage
-
val
uid: String
- Definition Classes
- CNTKModel → BasicLogging → Identifiable
-
val
ver: String
- Definition Classes
- BasicLogging
-
def
write: MLWriter
- Definition Classes
- ComplexParamsWritable → MLWritable