com.intel.analytics.zoo.pipeline.nnframes
Construct a NNEstimator with a featurePreprocessing only.
Construct a NNEstimator with a featurePreprocessing only. The constructor is useful when both feature and label are derived from the same column of the original DataFrame.
BigDL module to be optimized
BigDL criterion method
A Preprocessing that transforms the feature data to a Sample[T].
Construct a NNEstimator with a feature Preprocessing and label Preprocessing.
Construct a NNEstimator with a feature Preprocessing and label Preprocessing.
BigDL module to be optimized
BigDL criterion method
Preprocessing[Any, Tensor[T] ]
Preprocessing[Any, Tensor[T] ]
Construct a NNEstimator with a feature size and label size.
Construct a NNEstimator with a feature size and label size. The constructor is useful when the feature column and label column contains the following data types: Float, Double, Int, Array[Float], Array[Double], Array[Int] and MLlib Vector. The feature and label data are converted to Tensors with the specified sizes before sending to the model.
BigDL module to be optimized
BigDL criterion method
The size (Tensor dimensions) of the feature data. e.g. an image may be with width * height = 28 * 28, featureSize = Array(28, 28).
The size (Tensor dimensions) of the label data.
Construct a NNEstimator with default Preprocessing: A SeqToTensor
Construct a NNEstimator with default Preprocessing: A SeqToTensor
BigDL module to be optimized
BigDL criterion method