fix: correct text sentiment analysis sample (#222)

dizcology · web-flow · commit 0befde36bfd4 · 2021-02-12T12:18:17.000-08:00
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 = {
