Eseguire la previsione delle serie temporali

Seleziona una versione della documentazione:

In questa pagina viene descritto come eseguire previsioni di serie temporali direttamente nel database AlloyDB Omni. Puoi utilizzare vari modelli di previsione, tra cui i modelli TimesFM, per aiutarti a gestire l'incertezza e prendere decisioni strategiche consapevoli. Per maggiori dettagli sul modello, consulta il blog di ricerca di TimesFM.

La funzione ai.forecast di AlloyDB Omni è progettata per funzionare con qualsiasi modello registrato con il tipo di modello ts_forecasting, offrendoti la flessibilità di utilizzare il modello migliore per le tue esigenze. Questo documento fornisce una guida dettagliata sull'utilizzo di TimesFM, ma puoi applicare i principi anche ad altri modelli.

Prima di iniziare

Prima di poter eseguire la previsione delle serie temporali, devi attivare il flag di database di previsione in AlloyDB Omni e registrare un modello di previsione.

Registra un modello di previsione

Con AlloyDB Omni puoi utilizzare vari modelli di previsione.

Questa sezione fornisce istruzioni per la registrazione del modello TimesFM. Per istruzioni sulla registrazione di altri modelli, consulta Registra un endpoint del modello con la gestione degli endpoint del modello.

Utilizza il modello TimesFM

I modelli TimesFM sono disponibili in due versioni:

  • TimesFM 1.0: la versione iniziale del modello TimesFM.
  • TimesFM 2.0: questa versione contiene un checkpoint con 500 milioni di parametri e il relativo codice di inferenza. Offre prestazioni migliori rispetto al modello 1.0.
  • TimesFM 2.5: questa versione contiene un checkpoint con 200 milioni di parametri e il relativo codice di inferenza. Offre prestazioni del modello migliori rispetto al modello 2.0.

Tutti i modelli sono disponibili in Model Garden per un deployment rapido.

Per utilizzare i modelli TimesFM, devi prima eseguire il deployment del modello TimesFM in un endpoint Vertex AI, quindi registrarlo in AlloyDB Omni.

Esegui il deployment del modello TimesFM su un endpoint Vertex AI

Puoi eseguire il deployment del modello con un solo clic dalla console Google Cloud o manualmente utilizzando un notebook Colaboratory.

Deployment con un solo clic

Per eseguire il deployment del modello TimesFM utilizzando un deployment con un solo clic, segui questi passaggi:

  1. Nella console Google Cloud , vai alla pagina Model Garden:

    Vai a Model Garden
  2. Nel campo Cerca, inserisci TimesFM.
  3. Nei risultati di ricerca, fai clic su TimesFM.
  4. Nella pagina TimesFM, completa i seguenti passaggi:

    1. Fai clic su Esegui il deployment del modello.
    2. Seleziona Vertex AI.
  5. Nel riquadro Esegui il deployment su Vertex AI, configura ed esegui il deployment del modello:

    1. Per utilizzare l'ultima versione del modello TimesFM, seleziona google/timesfm-2.0 nel menu a discesa ID risorsa.
    2. Per semplificare l'identificazione del deployment in un secondo momento, inserisci quanto segue:
      1. In Nome modello, un MODEL_NAME univoco.
      2. In Nome endpoint, un ENDPOINT_NAME univoco.
    3. Per espandere le opzioni di configurazione, in Impostazioni di deployment, fai clic su Avanzate.
      1. Per garantire la latenza più bassa possibile, nel menu a discesa Regione, seleziona la stessa regione in cui si trova l'istanza AlloyDB Omni.
      2. Per ottenere le prestazioni migliori, nel menu a discesa Specifica della macchina, seleziona una specifica della macchina appropriata. Tieni presente che una GPU o una TPU potrebbe offrire una latenza inferiore.
    4. Nel menu a discesa Accesso endpoint di Policy di disponibilità, seleziona Pubblico (endpoint condiviso).

      In questo modo viene creato un endpoint nel dominio aiplatform che funziona con AlloyDB Omni.

    5. Per eseguire il deployment del modello su un endpoint gestito, fai clic su Esegui il deployment. Il deployment potrebbe richiedere circa 10-15 minuti.
  6. Dopo il deployment del modello TimesFM, in Endpoint viene visualizzato un link simile a google_timesfm-VERSION-one-click-deploy.
  7. Per ottenere l'URL della richiesta del modello, fai clic sul link di deployment in Endpoint.
  8. Nella pagina dei dettagli dell'endpoint, fai clic su Richiesta di esempio.
  9. Nel riquadro Richiesta di esempio, individua e prendi nota dell'URL della richiesta del modello. Utilizza queste informazioni in Registra il modello TimesFM in AlloyDB Omni, ad esempio https://us-central1-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/REGION/endpoints/ENDPOINT_ID:predict.

    Assicurati che l'URL della richiesta contenga aiplatform.googleapis.com.

Deployment manuale

Per eseguire manualmente il deployment del modello TimesFM utilizzando un blocco note Colaboratory, segui questi passaggi:

  1. Vai alla pagina Model Garden nella console Google Cloud :

    Vai a Model Garden
  2. Nel campo Cerca, inserisci TimesFM.
  3. Nei risultati di ricerca, fai clic su TimesFM per aprire la scheda del modello.
  4. Nella pagina TimesFM, fai clic su CO Apri blocco note.
  5. Nella finestra Apri notebook, fai clic sul link Colab Enterprise per TimesFM 2.0.
  6. Nella pagina Collab Enterprise, completa i seguenti passaggi:

    1. Per stabilire una connessione all'ambiente di runtime, fai clic su Connetti.
    2. (Facoltativo) Se necessario, completa i seguenti passaggi:

      • Nella sezione Prerequisiti di Configura il progetto Google Cloud, imposta BUCKET_URI e REGION, ad esempio imposta BUCKET_URI su gs://your-unique-bucket-name e REGION su us-central1.

      • Nel menu a discesa accelerator_type di Esegui il deployment di TimesFM su un endpoint Vertex AI, seleziona un tipo di acceleratore diverso, se necessario.
      • In Passaggio 3: imposta l'orizzonte di previsione massimo, compila i campi horizon e max_context, ad esempio imposta horizon su 128 e max_context su 512.
    3. Deseleziona la casella di controllo use_dedicated_endpoint.
    4. Esegui tutte le celle nel notebook.

    L'output del notebook fornisce i dettagli dell'endpoint nel seguente formato: projects/PROJECT_ID/locations/REGION/endpoints/ENDPOINT_ID.

  7. Per creare model_request_url per la chiamata di creazione del modello SQL, sostituisci gli identificatori del progetto e dell'endpoint con i dettagli dell'endpoint del passaggio precedente come segue:

    CALL google_ml.create_model(
     model_id => 'timesfm_v2',
     model_qualified_name => 'timesfm_v2',
     model_type => 'ts_forecasting',
     model_provider => 'google',
     model_request_url => 'https://REGION-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/REGION/endpoints/ENDPOINT_ID:predict');

    Questo model_request_url è necessario nella chiamata google_ml.create_model quando registri il modello TimesFM in AlloyDB Omni, come descritto nella sezione seguente.

  8. Per verificare che la funzione ai.forecast funzioni come previsto, esegui la seguente query di esempio:

    SELECT * FROM ai.forecast(
       source_query=> '
         SELECT my_timestamp, my_data_value FROM (
           VALUES
             (CAST(''2025-01-01'' AS TIMESTAMP), 10.0),
             (CAST(''2025-01-02'' AS TIMESTAMP), 12.0),
             (CAST(''2025-01-03'' AS TIMESTAMP), 11.0),
             (CAST(''2025-01-04'' AS TIMESTAMP), 13.0),
             (CAST(''2025-01-05'' AS TIMESTAMP), 15.0),
             (CAST(''2025-01-06'' AS TIMESTAMP), 14.0),
             (CAST(''2025-01-07'' AS TIMESTAMP), 16.0),
             (CAST(''2025-01-08'' AS TIMESTAMP), 18.0),
             (CAST(''2025-01-09'' AS TIMESTAMP), 17.0),
             (CAST(''2025-01-10'' AS TIMESTAMP), 20.0)
           ) AS t (my_timestamp, my_data_value)
         ' ,
       model_id => 'timesfm_v2',
       data_col => 'my_data_value',
       timestamp_col => 'my_timestamp',
    
       -- The rest of the parameters are the same
       horizon => 7,
       conf_level => 0.80
    );
Registra il modello TimesFM in AlloyDB Omni

Per registrare il modello TimesFM in AlloyDB Omni, segui questi passaggi:

  1. Verifica che l'estensione google_ml_integration sia installata nel database AlloyDB Omni che contiene i dati su cui vuoi eseguire le previsioni.

  2. Verifica di aver installato la versione 1.4.5 o successiva dell'estensione google_ml_integration.

    SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';
    

    Di seguito è riportato l'output di esempio:

    extversion
    ------------
    1.4.5
    (1 row)
    
  3. Chiama google_ml.create_model utilizzando l'URL della richiesta del modello che hai annotato in Registra un modello di previsione.

    CALL
    google_ml.create_model(
      model_id => 'MODEL_ID',
      model_qualified_name => 'MODEL_QUALIFIED_NAME',
      model_type => 'ts_forecasting',
      model_provider => 'google',
      model_request_url => 'https://REGION-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/REGION/endpoints/ENDPOINT_ID:predict' -- Example endpoint from Model Garden
      );
    

    Sostituisci quanto segue:

    • MODEL_ID: un identificatore univoco del modello registrato che vuoi utilizzare per la previsione, ad esempio vertex_timesfm.
    • MODEL_QUALIFIED_NAME: un nome definito dall'utente per il modello, ad esempio timesfm_v2.
    • REGION: la regione in cui si trova l'istanza, ad esempio us-central1.
    • PROJECT_ID: il nome del progetto in cui è stato eseguito il deployment del modello, ad esempio forecast_project.
    • ENDPOINT_ID: il nome del deployment del modello, ad esempio my-timesfm-endpoint.

      model_request_url viene fornito da Vertex AI dopo il deployment del modello. Copia l'intero URL dal riquadro Richiesta di esempio nella console Vertex AI. Contiene già le informazioni corrette sul progetto e sull'endpoint.

Attiva il flag di database per la previsione

Prima di poter generare previsioni utilizzando il modello registrato, devi impostare il flag di database google_ml_integration.enable_forecasting su on.

Puoi attivare questo flag per la sessione corrente o impostarlo in modo che rimanga attivo per un database o un utente specifico tra le sessioni. Per impostare il flag di database, seleziona una delle seguenti opzioni:

  • Attiva il flag per la sessione corrente:

    SET google_ml_integration.enable_forecasting = on;
    
  • Attiva il flag per un database specifico:

    ALTER DATABASE DATABASE_NAME SET google_ml_integration.enable_forecasting = 'on';
    

    Sostituisci DATABASE_NAME con il nome del database.

  • Attiva il flag per un utente di database specifico:

    ALTER ROLE postgres SET google_ml_integration.enable_forecasting = 'on';
    

Genera previsioni

Dopo aver registrato il modello TimesFM e abilitato il flag necessario, puoi generare previsioni passando una tabella di origine o una query con i dati delle serie temporali.

La funzione ai.forecast è documentata separatamente. Per un elenco completo dei parametri, incluse le relative descrizioni ed esempi, consulta il riferimento per la gestione degli endpoint del modello.

Genera previsioni da una tabella di origine

L'esempio seguente mostra come chiamare la funzione ai.forecast utilizzando una tabella di origine per generare previsioni:

SELECT * FROM ai.forecast(
    model_id => 'MODEL_ID',
    source_table => 'time_series_data',
    data_col => 'data_points',
    timestamp_col => 'timestamp',
    horizon => 2,
    conf_level => 0.80
);

Genera previsioni da una query

L'esempio seguente mostra come generare previsioni chiamando la funzione ai.forecast utilizzando una sottoquery come origine dati:

SELECT * FROM ai.forecast(
    model_id => 'MODEL_ID',
    source_query => '(SELECT * FROM time_series_data ORDER BY timestamp LIMIT 1) AS time_series_data',
    data_col => 'data_points',
    timestamp_col => 'timestamp',
    horizon => 2,
    conf_level => 0.80
);

Utilizza altri modelli di previsione

Puoi utilizzare la funzione ai.forecast con qualsiasi altro modello di previsione delle serie temporali che registri in AlloyDB Omni. La chiave è assicurarsi che il modello sia registrato con il tipo di modello ts_forecasting.

Per istruzioni dettagliate su come registrare diversi tipi di modelli, consulta Registra e chiama modelli di AI remoti utilizzando la gestione degli endpoint dei modelli. Una volta registrato il modello, puoi utilizzarlo con la funzione ai.forecast specificando il relativo model_id.

Passaggi successivi