Inherited from MLWritable
Inherited from NNParams[T]
Inherited from HasBatchSize
Inherited from HasPredictionCol
Inherited from HasPredictionCol
Inherited from HasFeaturesCol
Inherited from HasFeaturesCol
Inherited from DLTransformerBase[NNModel[T]]
Inherited from Transformer
Inherited from PipelineStage
Inherited from Logging
Inherited from Params
Inherited from Serializable
Inherited from Serializable
Inherited from Identifiable
Inherited from AnyRef
Inherited from Any
NNModel extends Spark ML Transformer and supports BigDL model with Spark DataFrame data.
NNModel supports different feature data type through Preprocessing. We provide pre-defined Preprocessing for popular data types like Array or Vector in package com.intel.analytics.zoo.feature, while user can also develop customized Preprocessing. During transform, NNModel will extract feature data from input DataFrame and use the Preprocessing to prepare data for the model.
After transform, the prediction column contains the output of the model as Array[T], where T (Double or Float) is decided by the model type.