feat(automl): expose disable_early_stopping option for create_model

Helin Wang · Helin Wang · commit 79918bb8700d · 2019-11-12T12:57:17.000-08:00
Disable early stopping is turned off by default. It's defined in proto here: https://github.com/googleapis/google-cloud-python/blob/bfb4da8542981d2eedffe20f64e87ab528a17592/automl/google/cloud/automl_v1beta1/proto/tables.proto#L196 This PR exposes disable_early_stopping option for TablesClient.create_model.
diff --git a/automl/google/cloud/automl_v1beta1/tables/tables_client.py b/automl/google/cloud/automl_v1beta1/tables/tables_client.py
@@ -2109,6 +2109,7 @@ def create_model(
         model_metadata={},
         include_column_spec_names=None,
         exclude_column_spec_names=None,
+        disable_early_stopping=False,
         **kwargs
     ):
         """Create a model. This will train your model on the given dataset.
@@ -2168,6 +2169,10 @@ def create_model(
             exclude_column_spec_names(Optional[str]):
                 The list of the names of the columns you want to exclude and
                 not train your model on.
+            disable_early_stopping(Optional[bool]):
+                True if disable early stopping. By default, the early stopping
+                feature is enabled, which means that AutoML Tables might stop
+                training before the entire training budget has been used.
         Returns:
             google.api_core.operation.Operation:
                 An operation future that can be used to check for
@@ -2212,6 +2217,8 @@ def create_model(
         model_metadata["train_budget_milli_node_hours"] = train_budget_milli_node_hours
         if optimization_objective is not None:
             model_metadata["optimization_objective"] = optimization_objective
+        if disable_early_stopping:
+            model_metadata["disable_early_stopping"] = True
 
         dataset_id = dataset_name.rsplit("/", 1)[-1]
         columns = [
