From d271b5d738ede6be6d3f17d64268efc14de75bc3 Mon Sep 17 00:00:00 2001 From: henrylee Date: Tue, 21 May 2024 17:05:09 +0000 Subject: [PATCH] fix: Use Alphabet 10k sample dataset. --- ai-platform/snippets/embedding-model-tuning.js | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ai-platform/snippets/embedding-model-tuning.js b/ai-platform/snippets/embedding-model-tuning.js index b8775326be..153b89b362 100644 --- a/ai-platform/snippets/embedding-model-tuning.js +++ b/ai-platform/snippets/embedding-model-tuning.js @@ -24,10 +24,10 @@ async function main( pipelineJobDisplayName = 'embedding-customization-pipeline-sample', baseModelVersionId = 'text-embedding-004', taskType = 'DEFAULT', - queriesPath = 'gs://embedding-customization-pipeline/dataset/queries.jsonl', - corpusPath = 'gs://embedding-customization-pipeline/dataset/corpus.jsonl', - trainLabelPath = 'gs://embedding-customization-pipeline/dataset/train.tsv', - testLabelPath = 'gs://embedding-customization-pipeline/dataset/test.tsv', + corpusPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/corpus.jsonl', + queriesPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/queries.jsonl', + trainLabelPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/train.tsv', + testLabelPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/test.tsv', outputDimensionality = 768, learningRateMultiplier = 1.0, batchSize = 128,