fix: Use Alphabet 10k sample dataset. (#9337)

GoogleCloudPlatform · May 23, 2024 · 84fed3d · 84fed3d
1 parent b770340
commit 84fed3d
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/aiplatform/src/test/java/aiplatform/EmbeddingModelTuningSampleTest.java b/aiplatform/src/test/java/aiplatform/EmbeddingModelTuningSampleTest.java
@@ -56,12 +56,14 @@ public class EmbeddingModelTuningSampleTest {
   private static final String BASE_MODEL_VERSION_ID = "text-embedding-004";
   private static final String TASK_TYPE = "DEFAULT";
   private static final String JOB_DISPLAY_NAME = "embedding-customization-pipeline-sample";
-  private static final String QUERIES =
-      "gs://embedding-customization-pipeline/dataset/queries.jsonl";
-  private static final String CORPUS = "gs://embedding-customization-pipeline/dataset/corpus.jsonl";
+  private static final String CORPUS = 
+      "gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/corpus.jsonl";
+  private static final String QUERIES = 
+      "gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/queries.jsonl";
   private static final String TRAIN_LABEL =
-      "gs://embedding-customization-pipeline/dataset/train.tsv";
-  private static final String TEST_LABEL = "gs://embedding-customization-pipeline/dataset/test.tsv";
+      "gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/train.tsv";
+  private static final String TEST_LABEL = 
+      "gs://cloud-samples-data/ai-platform/embedding/goog-10k-2024/r11/test.tsv";
   private static final String OUTPUT_DIR =
       "gs://ucaip-samples-us-central1/training_pipeline_output";
   private static final double LEARNING_RATE_MULTIPLIER = 0.3;