com.intel.analytics.zoo.pipeline.nnframes
Construct a NNClassifierModel with a feature Preprocessing.
Construct a NNClassifierModel with a feature Preprocessing.
BigDL module to be optimized
Preprocessing[F, Tensor[T] ].
Construct a NNClassifierModel with sizes of multiple model inputs.
Construct a NNClassifierModel with sizes of multiple model inputs. The constructor is useful when the feature column contains the following data types: Float, Double, Int, Array[Float], Array[Double], Array[Int] and MLlib Vector. The feature data are converted to Tensors with the specified sizes before sending to the model.
This API is used for multi-input model, where user need to specify the tensor sizes for each of the model input.
model to be used, which should be a multi-input model.
The sizes (Tensor dimensions) of the feature data.
Construct a NNClassifierModel with a feature size.
Construct a NNClassifierModel with a feature size. The constructor is useful when the feature column contains the following data types: Float, Double, Int, Array[Float], Array[Double], Array[Int] and MLlib Vector. The feature data are converted to Tensors with the specified sizes before sending to the model.
BigDL module to be optimized
The size (Tensor dimensions) of the feature data. e.g. an image may be with width * height = 28 * 28, featureSize = Array(28, 28).
Construct a NNClassifierModel with default Preprocessing, SeqToTensor
Construct a NNClassifierModel with default Preprocessing, SeqToTensor
BigDL module to be optimized