BigDL module to be optimized
BigDL criterion method
The size (Tensor dimensions) of the feature data.
BigDL criterion method
BigDL criterion method
The size (Tensor dimensions) of the feature data.
The size (Tensor dimensions) of the feature data.
learning rate for the optimizer in the DLEstimator.
learning rate for the optimizer in the DLEstimator. Default: 0.001
learning rate decay.
learning rate decay. Default: 0
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
BigDL module to be optimized
BigDL module to be optimized
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
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
DLClassifier is a specialized DLEstimator that simplifies the data format for classification tasks. It only supports label column of DoubleType. and the fitted DLClassifierModel will have the prediction column of DoubleType.