com.intel.analytics.zoo.models.recommendation
The factory method to create a WideAndDeep instance.
The factory method to create a WideAndDeep instance.
String. "wide", "deep", "wide_n_deep" are supported. Default is "wide_n_deep".
The number of classes. Positive integer.
An instance of ColumnFeatureInfo.
Units of hidden layers for the deep model. Array of positive integers. Default is Array(40, 20, 10).
Load an existing WideAndDeep model (with weights).
Load an existing WideAndDeep model (with weights).
Numeric type of parameter(e.g. weight, bias). Only support float/double now.
The path for the pre-defined model. Local file system, HDFS and Amazon S3 are supported. HDFS path should be like "hdfs://[host]:[port]/xxx". Amazon S3 path should be like "s3a://bucket/xxx".
The path for pre-trained weights if any. Default is null.