https://github.com/GoogleCloudPlatform/mlops-with-vertex-ai/issues/22

oonisim · Jun 13, 2022 · ccde346 · ccde346
1 parent 44d9107
commit ccde346
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/courses/mlops-with-vertex-ai/src/model_training/model.py b/courses/mlops-with-vertex-ai/src/model_training/model.py
@@ -57,6 +57,10 @@ def _create_binary_classifier(feature_vocab_sizes, hyperparams):
             layers.append(embedding_output)
         elif feature_name in features.ONEHOT_CATEGORICAL_FEATURE_NAMES:
             vocab_size = feature_vocab_sizes[feature_name]
+
+            # --------------------------------------------------------------------------------
+            # https://github.com/GoogleCloudPlatform/mlops-with-vertex-ai/issues/22
+            # --------------------------------------------------------------------------------
             # onehot_layer = keras.layers.experimental.preprocessing.CategoryEncoding(
             #     max_tokens=vocab_size,
             #     output_mode="binary",