Skip to content

Commit d271b5d

Browse files
lee1premium-2024iennae
authored andcommitted
fix: Use Alphabet 10k sample dataset.
1 parent a0427e5 commit d271b5d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ai-platform/snippets/embedding-model-tuning.js

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,10 @@ async function main(
2424
pipelineJobDisplayName = 'embedding-customization-pipeline-sample',
2525
baseModelVersionId = 'text-embedding-004',
2626
taskType = 'DEFAULT',
27-
queriesPath = 'gs://embedding-customization-pipeline/dataset/queries.jsonl',
28-
corpusPath = 'gs://embedding-customization-pipeline/dataset/corpus.jsonl',
29-
trainLabelPath = 'gs://embedding-customization-pipeline/dataset/train.tsv',
30-
testLabelPath = 'gs://embedding-customization-pipeline/dataset/test.tsv',
27+
corpusPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/corpus.jsonl',
28+
queriesPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/queries.jsonl',
29+
trainLabelPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/train.tsv',
30+
testLabelPath = 'gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/test.tsv',
3131
outputDimensionality = 768,
3232
learningRateMultiplier = 1.0,
3333
batchSize = 128,

0 commit comments

Comments
 (0)