Skip to content

Commit 643d335

Browse files
TheMichaelHuTheMichaelHu
authored
feat: add seq2seq forecasting training job (#1196)
Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly: - [x] Make sure to open an issue as a [bug/issue](https://github.com/googleapis/python-aiplatform/issues/new/choose) before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea - [x] Ensure the tests and linter pass - [x] Code coverage does not decrease (if any source code was changed) - [x] Appropriate docs were updated (if necessary) Fixes b/229909845 🦕 --- Adds a `SequenceToSequencePlusForecastingTrainingJob` to training jobs. This job has the exact same signature as `AutoMLForecastingTrainingJob`, but we are creating a separate job in case the two models diverge in the future. The logic for `AutoMLForecastingTrainingJob` has been moved to a new abstract base class `_ForecastingTrainingJob`. The only things that differ between the seq2seq and automl training jobs that extend it are the `model_type` and `training_task_definition`.
1 parent efaf6ed commit 643d335

5 files changed

Lines changed: 1890 additions & 1619 deletions

File tree

‎google/cloud/aiplatform/__init__.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@
6363
CustomPythonPackageTrainingJob,
6464
AutoMLTabularTrainingJob,
6565
AutoMLForecastingTrainingJob,
66+
SequenceToSequencePlusForecastingTrainingJob,
6667
AutoMLImageTrainingJob,
6768
AutoMLTextTrainingJob,
6869
AutoMLVideoTrainingJob,
@@ -116,6 +117,7 @@
116117
"Model",
117118
"ModelEvaluation",
118119
"PipelineJob",
120+
"SequenceToSequencePlusForecastingTrainingJob",
119121
"TabularDataset",
120122
"Tensorboard",
121123
"TensorboardExperiment",

‎google/cloud/aiplatform/schema.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ class definition:
2323
custom_task = "gs://google-cloud-aiplatform/schema/trainingjob/definition/custom_task_1.0.0.yaml"
2424
automl_tabular = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_tabular_1.0.0.yaml"
2525
automl_forecasting = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_time_series_forecasting_1.0.0.yaml"
26+
seq2seq_plus_forecasting = "gs://google-cloud-aiplatform/schema/trainingjob/definition/seq2seq_plus_time_series_forecasting_1.0.0.yaml"
2627
automl_image_classification = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_image_classification_1.0.0.yaml"
2728
automl_image_object_detection = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_image_object_detection_1.0.0.yaml"
2829
automl_text_classification = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_text_classification_1.0.0.yaml"

0 commit comments

Comments
 (0)