Addestra un modello con il flusso di lavoro tabulare per la previsione

Questa pagina mostra come addestrare un modello di previsione da un set di dati tabulare con il flusso di lavoro tabulare per la previsione.

Per saperne di più sugli account di servizio utilizzati da questo flusso di lavoro, consulta Account di servizio per Tabular Workflows.

Se ricevi un errore relativo alle quote durante l'esecuzione del flusso di lavoro tabulare per la previsione, potresti dover richiedere una quota più elevata. Per saperne di più, consulta Gestire le quote per i flussi di lavoro tabulari.

Il flusso di lavoro tabulare per la previsione non supporta l'esportazione dei modelli.

API Workflow

Questo flusso di lavoro utilizza le API seguenti:

• Vertex AI
• Dataflow
• Compute Engine
• Cloud Storage

Ottieni l'URI del risultato precedente dell'ottimizzazione degli iperparametri

Se hai già completato un'esecuzione del flusso di lavoro tabulare per la previsione, puoi utilizzare il risultato dell'ottimizzazione degli iperparametri dell'esecuzione precedente per risparmiare tempo e risorse di addestramento. Puoi trovare il risultato dell'ottimizzazione degli iperparametri precedente utilizzando la console Google Cloud o caricandolo in modo programmatico con l'API.

def get_task_detail(
  task_details: List[Dict[str, Any]], task_name: str
) -> List[Dict[str, Any]]:
  for task_detail in task_details:
      if task_detail.task_name == task_name:
          return task_detail

pipeline_task_details = job.gca_resource.job_detail.task_details

stage_1_tuner_task = get_task_detail(
    pipeline_task_details, "automl-forecasting-stage-1-tuner"
)
stage_1_tuning_result_artifact_uri = (
    stage_1_tuner_task.outputs["tuning_result_output"].artifacts[0].uri
)

Addestra un modello

Il seguente codice campione mostra come eseguire una pipeline di addestramento di un modello:

job = aiplatform.PipelineJob(
    ...
    template_path=template_path,
    parameter_values=parameter_values,
    ...
)
job.run(service_account=SERVICE_ACCOUNT)

Il parametro facoltativo service_account in job.run() consente di impostare l'account di servizio Vertex AI Pipelines su un account a tua scelta.

Vertex AI supporta i seguenti metodi per l'addestramento del modello:

• Time Series Dense Encoder (TiDE). Per utilizzare questo metodo di addestramento del modello, definisci i valori della pipeline e dei parametri utilizzando la seguente funzione:

  template_path, parameter_values = automl_forecasting_utils.get_time_series_dense_encoder_forecasting_pipeline_and_parameters(...)
  

• Trasformatore di fusione temporale (TFT). Per utilizzare questo metodo di addestramento del modello, definisci i valori della pipeline e dei parametri utilizzando la seguente funzione:

  template_path, parameter_values = automl_forecasting_utils.get_temporal_fusion_transformer_forecasting_pipeline_and_parameters(...)
  

• AutoML (L2L). Per utilizzare questo metodo di addestramento del modello, definisci i valori della pipeline e dei parametri utilizzando la seguente funzione:

  template_path, parameter_values = automl_forecasting_utils.get_learn_to_learn_forecasting_pipeline_and_parameters(...)
  

• Seq2Seq+. Per utilizzare questo metodo di addestramento del modello, definisci i valori della pipeline e dei parametri utilizzando la seguente funzione:

  template_path, parameter_values = automl_forecasting_utils.get_sequence_to_sequence_forecasting_pipeline_and_parameters(...)
  

Per scoprire di più, consulta Metodi di addestramento dei modelli.

I dati di addestramento possono essere un file CSV in Cloud Storage o una tabella in BigQuery.

Di seguito è riportato un sottoinsieme di parametri di addestramento del modello:

Trasformazioni

Puoi fornire una mappatura del dizionario delle risoluzioni automatiche o del tipo alle colonne delle funzionalità. I tipi supportati sono: automatico, numerico, categoriale, di testo e timestamp.

Il seguente codice fornisce una funzione helper per il completamento del parametro transformations. Dimostra inoltre come utilizzare questa funzione per applicare trasformazioni automatiche a un insieme di colonne definito da una variabile features.

def generate_transformation(
      auto_column_names: Optional[List[str]]=None,
      numeric_column_names: Optional[List[str]]=None,
      categorical_column_names: Optional[List[str]]=None,
      text_column_names: Optional[List[str]]=None,
      timestamp_column_names: Optional[List[str]]=None,
    ) -> List[Dict[str, Any]]:
    if auto_column_names is None:
      auto_column_names = []
    if numeric_column_names is None:
      numeric_column_names = []
    if categorical_column_names is None:
      categorical_column_names = []
    if text_column_names is None:
      text_column_names = []
    if timestamp_column_names is None:
      timestamp_column_names = []
    return {
        "auto": auto_column_names,
        "numeric": numeric_column_names,
        "categorical": categorical_column_names,
        "text": text_column_names,
        "timestamp": timestamp_column_names,
    }

transformations = generate_transformation(auto_column_names=features)

Per saperne di più sulle trasformazioni, consulta Tipi di dati e trasformazioni.

Opzioni di personalizzazione del flusso di lavoro

Puoi personalizzare il flusso di lavoro tabulare per la previsione definendo valori di argomenti da passare durante la definizione della pipeline. Puoi personalizzare il tuo flusso di lavoro nei seguenti modi:

• Configura hardware
• Salta ricerca architettura

Configura hardware

Il seguente parametro di addestramento del modello consente di configurare i tipi di macchine e il numero di macchine per l'addestramento. Questa opzione è ideale se hai un set di dati di grandi dimensioni e vuoi ottimizzare l'hardware della macchina.

Il codice seguente mostra come impostare il tipo di macchina n1-standard-8 per il nodo principale di TensorFlow e il tipo di macchina n1-standard-4 per il nodo valutatore TensorFlow:

worker_pool_specs_override = [
  {"machine_spec": {"machine_type": "n1-standard-8"}}, # override for TF chief node
  {},  # override for TF worker node, since it's not used, leave it empty
  {},  # override for TF ps node, since it's not used, leave it empty
  {
    "machine_spec": {
        "machine_type": "n1-standard-4" # override for TF evaluator node
    }
  }
]

Salta la ricerca dell'architettura

Il seguente parametro di addestramento del modello consente di eseguire la pipeline senza la ricerca dell'architettura e di fornire un set di iperparametri di un'esecuzione precedente della pipeline.

Passaggi successivi

• Scopri di più sulle previsioni batch per i modelli di previsione.
• Scopri di più sui prezzi per l'addestramento del modello.