AutoMLTrainModelOperator

Google

Creates Google Cloud AutoML model.

View Source

Last Updated: Jan. 11, 2021

Access Instructions

Install the Google provider package into your Airflow environment.

Import the module into your DAG file and instantiate it with your desired params.

Parameters

modeldictModel definition.
project_idstrID of the Google Cloud project where model will be created if None then default project_id is used.
locationstrThe location of the project.
retryOptional[google.api_core.retry.Retry]A retry object used to retry requests. If None is specified, requests will not be retried.
timeoutOptional[float]The amount of time, in seconds, to wait for the request to complete. Note that if retry is specified, the timeout applies to each individual attempt.
metadataOptional[Sequence[Tuple[str, str]]]Additional metadata that is provided to the method.
gcp_conn_idstrThe connection ID to use to connect to Google Cloud.
impersonation_chainUnion[str, Sequence[str]]Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

Documentation

Creates Google Cloud AutoML model.

See also

For more information on how to use this operator, take a look at the guide: Operations On Models

Was this page helpful?