Die ai.forecast-Funktion von AlloyDB Omni ist für die Verwendung mit jedem Modell konzipiert, das mit dem Modelltyp ts_forecasting registriert ist. So können Sie das Modell verwenden, das am besten zu Ihren Anforderungen passt. In diesem Dokument finden Sie eine detaillierte Anleitung zur Verwendung von TimesFM. Sie können die Grundsätze aber auch auf andere Modelle anwenden.
Hinweis
Bevor Sie Zeitreihenprognosen erstellen können, müssen Sie in AlloyDB Omni das Datenbankflag für Prognosen aktivieren und ein Prognosemodell registrieren.
Prognosemodell registrieren
Sie können verschiedene Prognosemodelle mit AlloyDB Omni verwenden.
In diesem Abschnitt finden Sie eine Anleitung zum Registrieren des TimesFM-Modells. Eine Anleitung zum Registrieren anderer Modelle finden Sie unter Modellendpunkt bei der Modellendpunktverwaltung registrieren.
TimesFM-Modell verwenden
Es gibt zwei Versionen von TimesFM-Modellen:
- TimesFM 1.0: Die erste Version des TimesFM-Modells.
- TimesFM 2.0: Diese Version enthält einen Checkpoint mit 500 Millionen Parametern und den zugehörigen Inferenzcode. Es bietet eine bessere Modellleistung als das Modell 1.0.
- TimesFM 2.5: Diese Version enthält einen Checkpoint mit 200 Millionen Parametern und den zugehörigen Inferenzcode. Es bietet eine bessere Modellleistung als das 2.0-Modell.
Alle Modelle sind in Model Garden für die schnelle Bereitstellung verfügbar.
Wenn Sie TimesFM-Modelle verwenden möchten, müssen Sie das TimesFM-Modell zuerst an einem Vertex AI-Endpunkt bereitstellen und es dann in AlloyDB Omni registrieren.
TimesFM-Modell in einem Vertex AI-Endpunkt bereitstellen
Sie können das Modell mit einem Klick über die Google Cloud Console oder manuell über ein Colaboratory-Notebook bereitstellen.
Bereitstellung mit nur einem Klick
So stellen Sie das TimesFM-Modell mit einem Klick bereit:
Rufen Sie in der Google Cloud Console die Seite Model Garden auf:
Zu Model Garden- Geben Sie im Feld Suchen
TimesFMein. - Klicken Sie in den Suchergebnissen auf
TimesFM. -
Führen Sie auf der Seite TimesFM die folgenden Schritte aus:
- Klicken Sie auf Modell bereitstellen.
- Wählen Sie Vertex AI aus.
Konfigurieren und stellen Sie Ihr Modell im Bereich In Vertex AI bereitstellen bereit:
- Wenn Sie die aktuelle Version des TimesFM-Modells verwenden möchten, wählen Sie im Drop-down-Menü Ressourcen-ID die Option
google/timesfm-2.0aus. - Damit Sie die Bereitstellung später leichter identifizieren können, geben Sie Folgendes ein:
- Geben Sie unter Modellname einen eindeutigen
MODEL_NAMEein. - Geben Sie unter Endpunktname einen eindeutigen
ENDPOINT_NAMEein.
- Geben Sie unter Modellname einen eindeutigen
- Klicken Sie unter Bereitstellungseinstellungen auf Erweitert, um die Konfigurationsoptionen zu maximieren.
- Um eine möglichst geringe Latenz zu erzielen, wählen Sie im Drop-down-Menü Region dieselbe Region aus, in der sich Ihre AlloyDB Omni-Instanz befindet.
- Für die beste Leistung sollten Sie im Drop-down-Menü Maschinenspezifikation eine geeignete Maschinenspezifikation auswählen. Eine GPU oder TPU kann eine niedrigere Latenz bieten.
Wählen Sie im Drop-down-Menü Endpunktzugriff unter Verfügbarkeitsrichtlinien die Option Öffentlich (gemeinsamer Endpunkt) aus.
Dadurch wird ein Endpunkt in der
aiplatform-Domain erstellt, der mit AlloyDB Omni funktioniert.- Klicken Sie auf Bereitstellen, um das Modell auf einem verwalteten Endpunkt bereitzustellen. Die Bereitstellung kann etwa 10 bis 15 Minuten dauern.
- Wenn Sie die aktuelle Version des TimesFM-Modells verwenden möchten, wählen Sie im Drop-down-Menü Ressourcen-ID die Option
- Nachdem das TimesFM-Modell bereitgestellt wurde, wird in Endpunkte ein Link ähnlich wie
google_timesfm-VERSION-one-click-deployangezeigt. - Klicken Sie in Endpunkte auf den Bereitstellungslink, um die Modellanfrage-URL zu erhalten.
- Klicken Sie auf der Endpunkt-Detailseite auf Beispielanfrage.
Suchen Sie im Bereich Beispielanfrage nach der URL der Modellanfrage und notieren Sie sie. Sie verwenden diese Informationen in TimesFM-Modell in AlloyDB Omni registrieren, z. B.
https://us-central1-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/REGION/endpoints/ENDPOINT_ID:predict.Prüfen Sie, ob die Anfrage-URL
aiplatform.googleapis.comenthält.
Manuelle Bereitstellung
So stellen Sie das TimesFM-Modell manuell mit einem Colaboratory-Notebook bereit:
-
Rufen Sie in der Google Cloud Console die Seite Model Garden auf:
Zu Model Garden - Geben Sie im Feld Suchen
TimesFMein. - Klicken Sie in den Suchergebnissen auf
TimesFM, um die Modellkarte zu öffnen. - Klicken Sie auf der Seite TimesFM auf CO Open Notebook.
- Klicken Sie im Fenster Notebook öffnen auf den Link Colab Enterprise für TimesFM 2.0.
-
Führen Sie auf der Seite Colab Enterprise die folgenden Schritte aus:
- Klicken Sie auf Verbinden, um eine Verbindung zur Laufzeitumgebung herzustellen.
-
Optional: Führen Sie bei Bedarf die folgenden Schritte aus:
-
Legen Sie im Abschnitt Voraussetzungen unter Google Cloud-Projekt einrichten die BUCKET_URI und REGION fest, z. B. BUCKET_URI auf
gs://your-unique-bucket-nameund REGION aufus-central1. - Wählen Sie im Drop-down-Menü accelerator_type von TimesFM in einem Vertex AI-Endpunkt bereitstellen bei Bedarf einen anderen Beschleunigertyp aus.
- Füllen Sie in Schritt 3: Maximalen Prognosezeitraum festlegen die Felder horizon und max_context aus. Setzen Sie beispielsweise horizon auf
128und max_context auf512.
-
- Entfernen Sie das Häkchen aus dem Kästchen
use_dedicated_endpoint. - Führen Sie alle Zellen im Notebook aus.
Die Ausgabe des Notebooks enthält die Endpunktdetails im folgenden Format:
projects/PROJECT_ID/locations/REGION/endpoints/ENDPOINT_ID. -
Um die
model_request_urlfür den SQL-Modellerstellungsaufruf zu erstellen, ersetzen Sie die Projekt- und Endpunkt-IDs durch die Endpunktdetails aus dem vorherigen Schritt: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');
Sie benötigen diese
model_request_urlimgoogle_ml.create_model-Aufruf, wenn Sie das TimesFM-Modell in AlloyDB Omni registrieren, wie im nächsten Abschnitt beschrieben. Führen Sie die folgende Beispielabfrage aus, um zu prüfen, ob die Funktion
ai.forecastwie erwartet funktioniert: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 );
TimesFM-Modell in AlloyDB Omni registrieren
So registrieren Sie das TimesFM-Modell in AlloyDB Omni:
Prüfen Sie, ob die
google_ml_integration-Erweiterung in der AlloyDB Omni-Datenbank installiert ist, die die Daten enthält, für die Sie Vorhersagen erstellen möchten.Prüfen Sie, ob Sie Version 1.4.5 oder höher der
google_ml_integration-Erweiterung installiert haben.SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';Hier ein Beispiel für die Ausgabe:
extversion ------------ 1.4.5 (1 row)Rufen Sie
google_ml.create_modelmit der Modellanfrage-URL auf, die Sie sich unter Prognosemodell registrieren notiert haben.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 );Ersetzen Sie Folgendes:
MODEL_ID: Eine eindeutige Kennung für das registrierte Modell, das Sie für die Prognose verwenden möchten, z. B.vertex_timesfm.MODEL_QUALIFIED_NAME: Ein benutzerdefinierter Name für das Modell, z. B.timesfm_v2.REGION: Die Region, in der sich die Instanz befindet, z. B.us-central1.PROJECT_ID: der Name des Projekts, in dem das Modell bereitgestellt wird, z. B.forecast_project.ENDPOINT_ID: Der Name der Modellbereitstellung, z. B.my-timesfm-endpoint.Die
model_request_urlwird von Vertex AI bereitgestellt, nachdem Sie Ihr Modell bereitgestellt haben. Kopieren Sie die vollständige URL aus dem Bereich Beispielanfrage in der Vertex AI-Konsole. Sie enthält bereits die richtigen Projekt- und Endpunktinformationen.
Flag für die Prognosedatenbank aktivieren
Bevor Sie mit dem registrierten Modell Prognosen erstellen können, müssen Sie das Datenbank-Flag google_ml_integration.enable_forecasting auf on setzen.
Sie können dieses Flag für Ihre aktuelle Sitzung aktivieren oder festlegen, dass es für eine bestimmte Datenbank oder einen bestimmten Nutzer sitzungsübergreifend beibehalten wird. Wählen Sie eine der folgenden Optionen aus, um das Datenbankflag festzulegen:
Aktivieren Sie das Flag für die aktuelle Sitzung:
SET google_ml_integration.enable_forecasting = on;So aktivieren Sie das Flag für eine bestimmte Datenbank:
ALTER DATABASE DATABASE_NAME SET google_ml_integration.enable_forecasting = 'on';Ersetzen Sie
DATABASE_NAMEdurch den Namen Ihrer Datenbank.So aktivieren Sie das Flag für einen bestimmten Datenbanknutzer:
ALTER ROLE postgres SET google_ml_integration.enable_forecasting = 'on';
Prognosen erstellen
Nachdem Sie das TimesFM-Modell registriert und das erforderliche Flag aktiviert haben, können Sie Vorhersagen generieren, indem Sie entweder eine Quelltabelle oder eine Abfrage mit Ihren Zeitreihendaten übergeben.
Die Funktion ai.forecast wird separat dokumentiert. Eine vollständige Liste der Parameter mit Beschreibungen und Beispielen finden Sie in der Referenz zur Verwaltung von Modellendpunkten.
Prognosen aus einer Quelltabelle erstellen
Im folgenden Beispiel wird gezeigt, wie die Funktion ai.forecast mit einer Quelltabelle aufgerufen wird, um Vorhersagen zu generieren:
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
);
Prognosen aus einer Abfrage generieren
Im folgenden Beispiel wird gezeigt, wie Sie Prognosen generieren, indem Sie die Funktion ai.forecast mit einer Unterabfrage als Datenquelle aufrufen:
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
);
Andere Prognosemodelle verwenden
Sie können die Funktion ai.forecast mit jedem anderen Zeitreihenprognosemodell verwenden, das Sie in AlloyDB Omni registrieren. Wichtig ist, dass Ihr Modell mit dem Modelltyp ts_forecasting registriert ist.
Eine detaillierte Anleitung zum Registrieren verschiedener Modelltypen finden Sie unter Remote-KI-Modelle mit der Modellendpunktverwaltung registrieren und aufrufen.
Sobald Ihr Modell registriert ist, können Sie es mit der Funktion ai.forecast verwenden, indem Sie die model_id angeben.
Nächste Schritte
- Modellendpunkt bei der Modellendpunktverwaltung registrieren
- Mit KI-basierten SQL-Operatoren abfragen