@@ -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