fix: add target_column docstring (#473)

googleapis · Jun 11, 2021 · c0543cd · c0543cd
1 parent 481d172
commit c0543cd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py
@@ -2583,6 +2583,8 @@ def run(
                 [google.cloud.aiplatform.v1beta1.TrainingPipeline.training_task_definition].
                 For tabular Datasets, all their data is exported to
                 training, to pick and choose from.
+            target_column (str):
+                Required. The name of the column values of which the Model is to predict.
             training_fraction_split (float):
                 Required. The fraction of the input data that is to be
                 used to train the Model. This is ignored if Dataset is not provided.
@@ -2698,6 +2700,8 @@ def _run(
                 [google.cloud.aiplatform.v1beta1.TrainingPipeline.training_task_definition].
                 For tabular Datasets, all their data is exported to
                 training, to pick and choose from.
+            target_column (str):
+                Required. The name of the column values of which the Model is to predict.
             training_fraction_split (float):
                 Required. The fraction of the input data that is to be
                 used to train the Model. This is ignored if Dataset is not provided.