fix: correct text sentiment analysis sample (#222)

googleapis · Feb 12, 2021 · 0befde3 · 0befde3
1 parent cdeb0ec
commit 0befde3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.sample_configs/param_handlers/create_training_pipeline_text_sentiment_analysis_sample.py b/.sample_configs/param_handlers/create_training_pipeline_text_sentiment_analysis_sample.py
@@ -18,7 +18,7 @@ def make_parent(parent: str) -> str:
 
 def make_training_pipeline(display_name: str, dataset_id: str, model_display_name: str) -> google.cloud.aiplatform_v1alpha1.types.training_pipeline.TrainingPipeline:
     # Use sentiment_max of 4
-    training_task_inputs_dict = {"sentiment_max": 4}
+    training_task_inputs_dict = {"sentiment_max": 10}
     training_task_inputs = to_protobuf_value(training_task_inputs_dict)
 
     training_pipeline = {

diff --git a/samples/snippets/create_training_pipeline_text_sentiment_analysis_sample.py b/samples/snippets/create_training_pipeline_text_sentiment_analysis_sample.py
@@ -32,7 +32,7 @@ def create_training_pipeline_text_sentiment_analysis_sample(
     client = aiplatform.gapic.PipelineServiceClient(client_options=client_options)
     # Use sentiment_max of 4
     training_task_inputs = trainingjob.definition.AutoMlTextSentimentInputs(
-        sentiment_max=4,
+        sentiment_max=10,
     ).to_value()
 
     training_pipeline = {