Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge


Advanced control of the model that most users won't need to use.

Used in the notebooks

Used in the tutorials

infer_prediction_signature Instantiate the model graph after training. This allows the model to be saved without specifying an input signature and without calling "predict", "evaluate". Disabling this logic can be useful in two situations: (1) When the exported signature is different from the one used during training, (2) When using a fixed-shape pre-processing that consume 1 dimensional tensors (as keras will automatically expend its shape to rank 2). For example, when using tf.Transform.
yggdrasil_training_config Yggdrasil Decision Forests training configuration. Expose a few extra hyper-parameters. yggdrasil_deployment_config: Configuration of the computing resources used to train the model e.g. number of threads. Does not impact the model quality.
yggdrasil_deployment_config A namedtuple alias for field number 2
fail_on_non_keras_compatible_feature_name If true (default), training will fail if one of the feature name is not compatible with part of the Keras API. If false, a warning will be generated instead.
predict_single_probability_for_binary_classification Only used for binary classification. If true (default), the prediction of a binary class model is a tensor of shape [None, 1] containing the probability of the positive class (value=1). If false, the prediction of a binary class model is a tensor of shape [None, num_classes=2] containing the probability of the complementary classes.
metadata_framework Metadata describing the framework used to train the model.
metadata_owner Metadata describing who trained the model.