DLEstimator

Instance Constructors

new DLEstimator(model: Module[T], criterion: Criterion[T], featureSize: Array[Int], labelSize: Array[Int], uid: String = "DLEstimator")(implicit arg0: ClassTag[T], ev: TensorNumeric[T])

model
BigDL module to be optimized
criterion
BigDL criterion method
featureSize
The size (Tensor dimensions) of the feature data. e.g. an image may be with width * height = 28 * 28, featureSize = Array(28, 28).
labelSize
The size (Tensor dimensions) of the label data.

Value Members

final def !=(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def !=(arg0: Any): Boolean

Definition Classes
Any
final def ##(): Int

Definition Classes
AnyRef → Any
final def $[T](param: Param[T]): T

Attributes
protected
Definition Classes
Params
final def ==(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def ==(arg0: Any): Boolean

Definition Classes
Any
final def asInstanceOf[T0]: T0

Definition Classes
Any
final val batchSize: Param[Int]

Definition Classes
HasBatchSize
final def clear(param: Param[_]): DLEstimator.this.type

Attributes
protected
Definition Classes
Params
def clone(): AnyRef

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( ... )
def copy(extra: ParamMap): DLEstimator[T]

Definition Classes
DLEstimator → DLEstimatorBase → Estimator → PipelineStage → Params
def copyValues[T <: Params](to: T, extra: ParamMap): T

Attributes
protected
Definition Classes
Params
val criterion: Criterion[T]

BigDL criterion method
final def defaultCopy[T <: Params](extra: ParamMap): T

Attributes
protected
Definition Classes
Params
final def eq(arg0: AnyRef): Boolean

Definition Classes
AnyRef
def equals(arg0: Any): Boolean

Definition Classes
AnyRef → Any
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 featureSize: Array[Int]

The size (Tensor dimensions) of the feature data.
The size (Tensor dimensions) of the feature data. e.g. an image may be with width * height = 28 * 28, featureSize = Array(28, 28).
final val featuresCol: Param[String]

Definition Classes
HasFeaturesCol
def finalize(): Unit

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( classOf[java.lang.Throwable] )
def fit(dataFrame: DataFrame): DLModel[T]

Definition Classes
DLEstimatorBase → Estimator
def fit(dataset: DataFrame, paramMaps: Array[ParamMap]): Seq[DLModel[T]]

Definition Classes
Estimator
def fit(dataset: DataFrame, paramMap: ParamMap): DLModel[T]

Definition Classes
Estimator
def fit(dataset: DataFrame, firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DLModel[T]

Definition Classes
Estimator
Annotations
@varargs()
final def get[T](param: Param[T]): Option[T]

Definition Classes
Params
def getBatchSize: Int

Definition Classes
HasBatchSize
final def getClass(): Class[_]

Definition Classes
AnyRef → Any
def getConvertFunc(colType: DataType): (Row, Int) ⇒ Seq[AnyVal]

Get conversion function to extract data from original DataFrame Default: 0
Get conversion function to extract data from original DataFrame Default: 0

Attributes
protected
Definition Classes
DLParams
final def getDefault[T](param: Param[T]): Option[T]

Definition Classes
Params
final def getFeaturesCol: String

Definition Classes
HasFeaturesCol
final def getLabelCol: String

Definition Classes
HasLabelCol
def getLearningRate: Double

Definition Classes
DLParams
def getLearningRateDecay: Double

Definition Classes
DLParams
def getMaxEpoch: Int

Definition Classes
DLParams
def getOptimMethod: OptimMethod[T]

Definition Classes
DLParams
final def getOrDefault[T](param: Param[T]): T

Definition Classes
Params
def getParam(paramName: String): Param[Any]

Definition Classes
Params
final def getPredictionCol: String

Definition Classes
HasPredictionCol
def getVectorSeq(row: Row, colType: DataType, index: Int): Seq[AnyVal]

Definition Classes
VectorCompatibility
final def hasDefault[T](param: Param[T]): Boolean

Definition Classes
Params
def hasParam(paramName: String): Boolean

Definition Classes
Params
def hashCode(): Int

Definition Classes
AnyRef → Any
def internalFit(dataFrame: DataFrame): DLModel[T]

Attributes
protected
Definition Classes
DLEstimator → DLEstimatorBase
final def isDefined(param: Param[_]): Boolean

Definition Classes
Params
final def isInstanceOf[T0]: Boolean

Definition Classes
Any
final def isSet(param: Param[_]): Boolean

Definition Classes
Params
def isTraceEnabled(): Boolean

Attributes
protected
Definition Classes
Logging
final val labelCol: Param[String]

Definition Classes
HasLabelCol
val labelSize: Array[Int]

The size (Tensor dimensions) of the label data.
final val learningRate: DoubleParam

learning rate for the optimizer in the DLEstimator.
learning rate for the optimizer in the DLEstimator. Default: 0.001

Definition Classes
DLParams
final val learningRateDecay: DoubleParam

learning rate decay for each iteration.
learning rate decay for each iteration. Default: 0

Definition Classes
DLParams
def log: Logger

Attributes
protected
Definition Classes
Logging
def logDebug(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logDebug(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logError(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logError(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logInfo(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logInfo(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logName: String

Attributes
protected
Definition Classes
Logging
def logTrace(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logTrace(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logWarning(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logWarning(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
final val maxEpoch: IntParam

number of max Epoch for the training, an epoch refers to a traverse over the training data Default: 100
number of max Epoch for the training, an epoch refers to a traverse over the training data Default: 100

Definition Classes
DLParams
val model: Module[T]

BigDL module to be optimized
final def ne(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def notify(): Unit

Definition Classes
AnyRef
final def notifyAll(): Unit

Definition Classes
AnyRef
final val optimMethod: Param[OptimMethod[T]]

optimization method to be used.
optimization method to be used. BigDL supports many optimization methods like Adam, SGD and LBFGS. Refer to package com.intel.analytics.bigdl.optim for all the options. Default: SGD

Definition Classes
DLParams
lazy val params: Array[Param[_]]

Definition Classes
Params
final val predictionCol: Param[String]

Definition Classes
HasPredictionCol
final def set(paramPair: ParamPair[_]): DLEstimator.this.type

Attributes
protected
Definition Classes
Params
final def set(param: String, value: Any): DLEstimator.this.type

Attributes
protected
Definition Classes
Params
final def set[T](param: Param[T], value: T): DLEstimator.this.type

Attributes
protected
Definition Classes
Params
def setBatchSize(value: Int): DLEstimator.this.type
final def setDefault(paramPairs: ParamPair[_]*): DLEstimator.this.type

Attributes
protected
Definition Classes
Params
final def setDefault[T](param: Param[T], value: T): DLEstimator.this.type

Attributes
protected
Definition Classes
Params
def setFeaturesCol(featuresColName: String): DLEstimator.this.type
def setLabelCol(labelColName: String): DLEstimator.this.type
def setLearningRate(value: Double): DLEstimator.this.type
def setLearningRateDecay(value: Double): DLEstimator.this.type
def setMaxEpoch(value: Int): DLEstimator.this.type
def setOptimMethod(value: OptimMethod[T]): DLEstimator.this.type
def setPredictionCol(value: String): DLEstimator.this.type
final def synchronized[T0](arg0: ⇒ T0): T0

Definition Classes
AnyRef
def toString(): String

Definition Classes
Identifiable → AnyRef → Any
def transformSchema(schema: StructType): StructType

Definition Classes
DLEstimator → PipelineStage
def transformSchema(schema: StructType, logging: Boolean): StructType

Attributes
protected
Definition Classes
PipelineStage
Annotations
@DeveloperApi()
val uid: String

Definition Classes
DLEstimator → Identifiable
val validVectorTypes: Seq[VectorUDT]

Definition Classes
VectorCompatibility
def validateDataType(schema: StructType, colName: String): Unit

Validate if feature and label columns are of supported data types.
Validate if feature and label columns are of supported data types. Default: 0

Attributes
protected
Definition Classes
DLParams
def validateParams(): Unit

Definition Classes
Params
final def wait(): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long, arg1: Int): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
def wrapBigDLModel(m: Module[T], featureSize: Array[Int]): DLModel[T]

sub classes can extend the method and return required model for different transform tasks
sub classes can extend the method and return required model for different transform tasks

Attributes
protected

class DLEstimator[T] extends DLEstimatorBase[DLEstimator[T], DLModel[T]] with DLParams[T]

Instance Constructors

new DLEstimator(model: Module[T], criterion: Criterion[T], featureSize: Array[Int], labelSize: Array[Int], uid: String = "DLEstimator")(implicit arg0: ClassTag[T], ev: TensorNumeric[T])

Value Members

final def !=(arg0: AnyRef): Boolean

final def !=(arg0: Any): Boolean

final def ##(): Int

final def $[T](param: Param[T]): T

final def ==(arg0: AnyRef): Boolean

final def ==(arg0: Any): Boolean

final def asInstanceOf[T0]: T0

final val batchSize: Param[Int]

final def clear(param: Param[_]): DLEstimator.this.type

def clone(): AnyRef

def copy(extra: ParamMap): DLEstimator[T]

def copyValues[T <: Params](to: T, extra: ParamMap): T

val criterion: Criterion[T]

final def defaultCopy[T <: Params](extra: ParamMap): T

final def eq(arg0: AnyRef): Boolean

def equals(arg0: Any): Boolean

def explainParam(param: Param[_]): String

def explainParams(): String

final def extractParamMap(): ParamMap

final def extractParamMap(extra: ParamMap): ParamMap

val featureSize: Array[Int]

final val featuresCol: Param[String]

def finalize(): Unit

def fit(dataFrame: DataFrame): DLModel[T]

def fit(dataset: DataFrame, paramMaps: Array[ParamMap]): Seq[DLModel[T]]

def fit(dataset: DataFrame, paramMap: ParamMap): DLModel[T]

def fit(dataset: DataFrame, firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DLModel[T]

final def get[T](param: Param[T]): Option[T]

def getBatchSize: Int

final def getClass(): Class[_]

def getConvertFunc(colType: DataType): (Row, Int) ⇒ Seq[AnyVal]

final def getDefault[T](param: Param[T]): Option[T]

final def getFeaturesCol: String

final def getLabelCol: String

def getLearningRate: Double

def getLearningRateDecay: Double

def getMaxEpoch: Int

def getOptimMethod: OptimMethod[T]

final def getOrDefault[T](param: Param[T]): T

def getParam(paramName: String): Param[Any]

final def getPredictionCol: String

def getVectorSeq(row: Row, colType: DataType, index: Int): Seq[AnyVal]

final def hasDefault[T](param: Param[T]): Boolean

def hasParam(paramName: String): Boolean

def hashCode(): Int

def internalFit(dataFrame: DataFrame): DLModel[T]

final def isDefined(param: Param[_]): Boolean

final def isInstanceOf[T0]: Boolean

final def isSet(param: Param[_]): Boolean

def isTraceEnabled(): Boolean

final val labelCol: Param[String]

val labelSize: Array[Int]

final val learningRate: DoubleParam

final val learningRateDecay: DoubleParam

def log: Logger

def logDebug(msg: ⇒ String, throwable: Throwable): Unit

def logDebug(msg: ⇒ String): Unit

def logError(msg: ⇒ String, throwable: Throwable): Unit

def logError(msg: ⇒ String): Unit

def logInfo(msg: ⇒ String, throwable: Throwable): Unit

def logInfo(msg: ⇒ String): Unit

def logName: String

def logTrace(msg: ⇒ String, throwable: Throwable): Unit

def logTrace(msg: ⇒ String): Unit

def logWarning(msg: ⇒ String, throwable: Throwable): Unit

def logWarning(msg: ⇒ String): Unit

final val maxEpoch: IntParam

val model: Module[T]

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

final val optimMethod: Param[OptimMethod[T]]

lazy val params: Array[Param[_]]

final val predictionCol: Param[String]

final def set(paramPair: ParamPair[_]): DLEstimator.this.type