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(dataset: 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
final def getBatchSize: Int

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

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

Definition Classes
Params
def getFeatureArrayCol: String

Attributes
protected
Definition Classes
DLParams
final def getFeaturesCol: String

Definition Classes
HasFeaturesCol
def getLabelArrayCol: String

Attributes
protected
Definition Classes
DLEstimatorBase
final def getLabelCol: String

Definition Classes
HasLabelCol
def getLearningRate: Double
def getLearningRateDecay: Double
def getMaxEpoch: Int
def getOptimMethod: OptimMethod[_]
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
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(featureAndLabel: RDD[(Seq[AnyVal], Seq[AnyVal])]): 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.
val learningRate: DoubleParam

learning rate for the optimizer in the DLEstimator.
learning rate for the optimizer in the DLEstimator. Default: 0.001
val learningRateDecay: DoubleParam

learning rate decay.
learning rate decay. Default: 0
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
val maxEpoch: IntParam

number of max Epoch for the training, an epoch refers to a traverse over the training data Default: 100
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
val optimMethod: Param[OptimMethod[_]]

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
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[_]): DLEstimator.this.type
def setPredictionCol(value: String): DLEstimator.this.type
def supportedTypesToSeq(row: Row, colType: DataType, index: Int): Seq[AnyVal]

Definition Classes
DLParams
final def synchronized[T0](arg0: ⇒ T0): T0

Definition Classes
AnyRef
def toArrayType(dataset: DataFrame): RDD[(Seq[AnyVal], Seq[AnyVal])]

Attributes
protected
Definition Classes
DLEstimatorBase
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
def validateParams(): Unit

Definition Classes
Params
def validateSchema(schema: StructType): Unit

Attributes
protected
Definition Classes
DLEstimatorBase → DLParams
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 with HasBatchSize

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(dataset: 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]

final def getBatchSize: Int

final def getClass(): Class[_]

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

def getFeatureArrayCol: String

final def getFeaturesCol: String

def getLabelArrayCol: String

final def getLabelCol: String

def getLearningRate: Double

def getLearningRateDecay: Double

def getMaxEpoch: Int

def getOptimMethod: OptimMethod[_]

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

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

final def getPredictionCol: String

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

def hasParam(paramName: String): Boolean

def hashCode(): Int

def internalFit(featureAndLabel: RDD[(Seq[AnyVal], Seq[AnyVal])]): 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]

val learningRate: DoubleParam

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

val maxEpoch: IntParam

val model: Module[T]

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

val optimMethod: Param[OptimMethod[_]]

lazy val params: Array[Param[_]]

final val predictionCol: Param[String]

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