Linear Combined Deep Neural Networks
Also known as wide-n-deep
estimators, these are estimators for
TensorFlow Linear and DNN joined models for regression.
dnn_linear_combined_regressor(model_dir = NULL,
linear_feature_columns = NULL, linear_optimizer = "Ftrl",
dnn_feature_columns = NULL, dnn_optimizer = "Adagrad",
dnn_hidden_units = NULL, dnn_activation_fn = "relu", dnn_dropout = NULL,
label_dimension = 1L, weight_column = NULL,
input_layer_partitioner = NULL, config = NULL)
dnn_linear_combined_classifier(model_dir = NULL,
linear_feature_columns = NULL, linear_optimizer = "Ftrl",
dnn_feature_columns = NULL, dnn_optimizer = "Adagrad",
dnn_hidden_units = NULL, dnn_activation_fn = "relu", dnn_dropout = NULL,
n_classes = 2L, weight_column = NULL, label_vocabulary = NULL,
input_layer_partitioner = NULL, config = NULL)
Arguments
model_dir | Directory to save the model parameters, graph, and so on. This can also be used to load checkpoints from the directory into a estimator to continue training a previously saved model. |
linear_feature_columns | The feature columns used by linear (wide) part of the model. |
linear_optimizer | Either the name of the optimizer to be used when training the model, or a TensorFlow optimizer instance. Defaults to the FTRL optimizer. |
dnn_feature_columns | The feature columns used by the neural network (deep) part in the model. |
dnn_optimizer | Either the name of the optimizer to be used when training the model, or a TensorFlow optimizer instance. Defaults to the Adagrad optimizer. |
dnn_hidden_units | An integer vector, indicating the number of hidden
units in each layer. All layers are fully connected. For example,
|
dnn_activation_fn | The activation function to apply to each layer. This can either be an
actual activation function (e.g. |
dnn_dropout | When not |
label_dimension | Number of regression targets per example. This is the
size of the last dimension of the labels and logits |
weight_column | A string, or a numeric column created by
|
input_layer_partitioner | An optional partitioner for the input layer.
Defaults to |
config | A run configuration created by |
n_classes | The number of label classes. |
label_vocabulary | A list of strings represents possible label values.
If given, labels must be string type and have any value in
|
See also
Other canned estimators: dnn_estimators
,
linear_estimators