@@ -5150,94 +5150,6 @@ class AutoMLForecastingTrainingJob(_ForecastingTrainingJob):
5150
5150
_training_task_definition = schema .training_job .definition .automl_forecasting
5151
5151
_supported_training_schemas = (schema .training_job .definition .automl_forecasting ,)
5152
5152
5153
- def run (
5154
- self ,
5155
- dataset : datasets .TimeSeriesDataset ,
5156
- target_column : str ,
5157
- time_column : str ,
5158
- time_series_identifier_column : str ,
5159
- unavailable_at_forecast_columns : List [str ],
5160
- available_at_forecast_columns : List [str ],
5161
- forecast_horizon : int ,
5162
- data_granularity_unit : str ,
5163
- data_granularity_count : int ,
5164
- training_fraction_split : Optional [float ] = None ,
5165
- validation_fraction_split : Optional [float ] = None ,
5166
- test_fraction_split : Optional [float ] = None ,
5167
- predefined_split_column_name : Optional [str ] = None ,
5168
- timestamp_split_column_name : Optional [str ] = None ,
5169
- weight_column : Optional [str ] = None ,
5170
- time_series_attribute_columns : Optional [List [str ]] = None ,
5171
- context_window : Optional [int ] = None ,
5172
- export_evaluated_data_items : bool = False ,
5173
- export_evaluated_data_items_bigquery_destination_uri : Optional [str ] = None ,
5174
- export_evaluated_data_items_override_destination : bool = False ,
5175
- quantiles : Optional [List [float ]] = None ,
5176
- validation_options : Optional [str ] = None ,
5177
- budget_milli_node_hours : int = 1000 ,
5178
- model_display_name : Optional [str ] = None ,
5179
- model_labels : Optional [Dict [str , str ]] = None ,
5180
- model_id : Optional [str ] = None ,
5181
- parent_model : Optional [str ] = None ,
5182
- is_default_version : Optional [bool ] = True ,
5183
- model_version_aliases : Optional [Sequence [str ]] = None ,
5184
- model_version_description : Optional [str ] = None ,
5185
- additional_experiments : Optional [List [str ]] = None ,
5186
- hierarchy_group_columns : Optional [List [str ]] = None ,
5187
- hierarchy_group_total_weight : Optional [float ] = None ,
5188
- hierarchy_temporal_total_weight : Optional [float ] = None ,
5189
- hierarchy_group_temporal_total_weight : Optional [float ] = None ,
5190
- window_column : Optional [str ] = None ,
5191
- window_stride_length : Optional [int ] = None ,
5192
- window_max_count : Optional [int ] = None ,
5193
- holiday_regions : Optional [List [str ]] = None ,
5194
- sync : bool = True ,
5195
- create_request_timeout : Optional [float ] = None ,
5196
- ) -> models .Model :
5197
- return super ().run (
5198
- dataset = dataset ,
5199
- target_column = target_column ,
5200
- time_column = time_column ,
5201
- time_series_identifier_column = time_series_identifier_column ,
5202
- unavailable_at_forecast_columns = unavailable_at_forecast_columns ,
5203
- available_at_forecast_columns = available_at_forecast_columns ,
5204
- forecast_horizon = forecast_horizon ,
5205
- data_granularity_unit = data_granularity_unit ,
5206
- data_granularity_count = data_granularity_count ,
5207
- training_fraction_split = training_fraction_split ,
5208
- validation_fraction_split = validation_fraction_split ,
5209
- test_fraction_split = test_fraction_split ,
5210
- predefined_split_column_name = predefined_split_column_name ,
5211
- timestamp_split_column_name = timestamp_split_column_name ,
5212
- weight_column = weight_column ,
5213
- time_series_attribute_columns = time_series_attribute_columns ,
5214
- context_window = context_window ,
5215
- budget_milli_node_hours = budget_milli_node_hours ,
5216
- export_evaluated_data_items = export_evaluated_data_items ,
5217
- export_evaluated_data_items_bigquery_destination_uri = export_evaluated_data_items_bigquery_destination_uri ,
5218
- export_evaluated_data_items_override_destination = export_evaluated_data_items_override_destination ,
5219
- quantiles = quantiles ,
5220
- validation_options = validation_options ,
5221
- model_display_name = model_display_name ,
5222
- model_labels = model_labels ,
5223
- model_id = model_id ,
5224
- parent_model = parent_model ,
5225
- is_default_version = is_default_version ,
5226
- model_version_aliases = model_version_aliases ,
5227
- model_version_description = model_version_description ,
5228
- additional_experiments = additional_experiments ,
5229
- hierarchy_group_columns = hierarchy_group_columns ,
5230
- hierarchy_group_total_weight = hierarchy_group_total_weight ,
5231
- hierarchy_temporal_total_weight = hierarchy_temporal_total_weight ,
5232
- hierarchy_group_temporal_total_weight = hierarchy_group_temporal_total_weight ,
5233
- window_column = window_column ,
5234
- window_stride_length = window_stride_length ,
5235
- window_max_count = window_max_count ,
5236
- holiday_regions = holiday_regions ,
5237
- sync = sync ,
5238
- create_request_timeout = create_request_timeout ,
5239
- )
5240
-
5241
5153
5242
5154
class SequenceToSequencePlusForecastingTrainingJob (_ForecastingTrainingJob ):
5243
5155
_model_type = "Seq2Seq"
@@ -5246,94 +5158,6 @@ class SequenceToSequencePlusForecastingTrainingJob(_ForecastingTrainingJob):
5246
5158
schema .training_job .definition .seq2seq_plus_forecasting ,
5247
5159
)
5248
5160
5249
- def run (
5250
- self ,
5251
- dataset : datasets .TimeSeriesDataset ,
5252
- target_column : str ,
5253
- time_column : str ,
5254
- time_series_identifier_column : str ,
5255
- unavailable_at_forecast_columns : List [str ],
5256
- available_at_forecast_columns : List [str ],
5257
- forecast_horizon : int ,
5258
- data_granularity_unit : str ,
5259
- data_granularity_count : int ,
5260
- training_fraction_split : Optional [float ] = None ,
5261
- validation_fraction_split : Optional [float ] = None ,
5262
- test_fraction_split : Optional [float ] = None ,
5263
- predefined_split_column_name : Optional [str ] = None ,
5264
- timestamp_split_column_name : Optional [str ] = None ,
5265
- weight_column : Optional [str ] = None ,
5266
- time_series_attribute_columns : Optional [List [str ]] = None ,
5267
- context_window : Optional [int ] = None ,
5268
- export_evaluated_data_items : bool = False ,
5269
- export_evaluated_data_items_bigquery_destination_uri : Optional [str ] = None ,
5270
- export_evaluated_data_items_override_destination : bool = False ,
5271
- quantiles : Optional [List [float ]] = None ,
5272
- validation_options : Optional [str ] = None ,
5273
- budget_milli_node_hours : int = 1000 ,
5274
- model_display_name : Optional [str ] = None ,
5275
- model_labels : Optional [Dict [str , str ]] = None ,
5276
- model_id : Optional [str ] = None ,
5277
- parent_model : Optional [str ] = None ,
5278
- is_default_version : Optional [bool ] = True ,
5279
- model_version_aliases : Optional [Sequence [str ]] = None ,
5280
- model_version_description : Optional [str ] = None ,
5281
- additional_experiments : Optional [List [str ]] = None ,
5282
- hierarchy_group_columns : Optional [List [str ]] = None ,
5283
- hierarchy_group_total_weight : Optional [float ] = None ,
5284
- hierarchy_temporal_total_weight : Optional [float ] = None ,
5285
- hierarchy_group_temporal_total_weight : Optional [float ] = None ,
5286
- window_column : Optional [str ] = None ,
5287
- window_stride_length : Optional [int ] = None ,
5288
- window_max_count : Optional [int ] = None ,
5289
- holiday_regions : Optional [List [str ]] = None ,
5290
- sync : bool = True ,
5291
- create_request_timeout : Optional [float ] = None ,
5292
- ) -> models .Model :
5293
- return super ().run (
5294
- dataset = dataset ,
5295
- target_column = target_column ,
5296
- time_column = time_column ,
5297
- time_series_identifier_column = time_series_identifier_column ,
5298
- unavailable_at_forecast_columns = unavailable_at_forecast_columns ,
5299
- available_at_forecast_columns = available_at_forecast_columns ,
5300
- forecast_horizon = forecast_horizon ,
5301
- data_granularity_unit = data_granularity_unit ,
5302
- data_granularity_count = data_granularity_count ,
5303
- training_fraction_split = training_fraction_split ,
5304
- validation_fraction_split = validation_fraction_split ,
5305
- test_fraction_split = test_fraction_split ,
5306
- predefined_split_column_name = predefined_split_column_name ,
5307
- timestamp_split_column_name = timestamp_split_column_name ,
5308
- model_id = model_id ,
5309
- parent_model = parent_model ,
5310
- is_default_version = is_default_version ,
5311
- model_version_aliases = model_version_aliases ,
5312
- model_version_description = model_version_description ,
5313
- weight_column = weight_column ,
5314
- time_series_attribute_columns = time_series_attribute_columns ,
5315
- context_window = context_window ,
5316
- budget_milli_node_hours = budget_milli_node_hours ,
5317
- export_evaluated_data_items = export_evaluated_data_items ,
5318
- export_evaluated_data_items_bigquery_destination_uri = export_evaluated_data_items_bigquery_destination_uri ,
5319
- export_evaluated_data_items_override_destination = export_evaluated_data_items_override_destination ,
5320
- quantiles = quantiles ,
5321
- validation_options = validation_options ,
5322
- model_display_name = model_display_name ,
5323
- model_labels = model_labels ,
5324
- additional_experiments = additional_experiments ,
5325
- hierarchy_group_columns = hierarchy_group_columns ,
5326
- hierarchy_group_total_weight = hierarchy_group_total_weight ,
5327
- hierarchy_temporal_total_weight = hierarchy_temporal_total_weight ,
5328
- hierarchy_group_temporal_total_weight = hierarchy_group_temporal_total_weight ,
5329
- window_column = window_column ,
5330
- window_stride_length = window_stride_length ,
5331
- window_max_count = window_max_count ,
5332
- holiday_regions = holiday_regions ,
5333
- sync = sync ,
5334
- create_request_timeout = create_request_timeout ,
5335
- )
5336
-
5337
5161
5338
5162
class AutoMLImageTrainingJob (_TrainingJob ):
5339
5163
_supported_training_schemas = (
0 commit comments