Skip to content

feat: add seq2seq forecasting training job #1196

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
gcf-merge-on-green merged 14 commits into from
Jun 3, 2022
Merged

feat: add seq2seq forecasting training job #1196

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
1c4ac24
Create abstract forecasting training job class.
TheMichaelHu May 4, 2022
26c70a2
Add a seq2seq training job.
TheMichaelHu May 9, 2022
7505165
fix context window description in _run
TheMichaelHu May 9, 2022
a1ba2c0
blacken
TheMichaelHu May 9, 2022
f6d3add
add supported schemas to forecasting base class
TheMichaelHu May 9, 2022
a4e2a33
Add seq2seq job to init file.
TheMichaelHu May 17, 2022
4e2cfaf
only keep super methods that need documentation
TheMichaelHu May 23, 2022
a084a21
add seq2seq e2e tests
TheMichaelHu May 27, 2022
4af79c7
foo
TheMichaelHu May 27, 2022
d68e530
blacken
TheMichaelHu May 28, 2022
8302756
fix merge conflict issues
TheMichaelHu May 28, 2022
6022bba
make e2e test parameterized
TheMichaelHu Jun 3, 2022
a44d055
fix prediction format issue
TheMichaelHu Jun 3, 2022
e5a9caa
blacken
TheMichaelHu Jun 3, 2022
File filter

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions google/cloud/aiplatform/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@
CustomPythonPackageTrainingJob,
AutoMLTabularTrainingJob,
AutoMLForecastingTrainingJob,
SequenceToSequencePlusForecastingTrainingJob,
AutoMLImageTrainingJob,
AutoMLTextTrainingJob,
AutoMLVideoTrainingJob,
Expand Down Expand Up @@ -116,6 +117,7 @@
"Model",
"ModelEvaluation",
"PipelineJob",
"SequenceToSequencePlusForecastingTrainingJob",
"TabularDataset",
"Tensorboard",
"TensorboardExperiment",
Expand Down
1 change: 1 addition & 0 deletions google/cloud/aiplatform/schema.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ class definition:
custom_task = "gs://google-cloud-aiplatform/schema/trainingjob/definition/custom_task_1.0.0.yaml"
automl_tabular = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_tabular_1.0.0.yaml"
automl_forecasting = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_time_series_forecasting_1.0.0.yaml"
seq2seq_plus_forecasting = "gs://google-cloud-aiplatform/schema/trainingjob/definition/seq2seq_plus_time_series_forecasting_1.0.0.yaml"
automl_image_classification = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_image_classification_1.0.0.yaml"
automl_image_object_detection = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_image_object_detection_1.0.0.yaml"
automl_text_classification = "gs://google-cloud-aiplatform/schema/trainingjob/definition/automl_text_classification_1.0.0.yaml"
Expand Down
Loading