Fix golden model name

PiperOrigin-RevId: 583001307
tensorflow · Nov 16, 2023 · 74bec4b · 74bec4b
1 parent a72a9d6
commit 74bec4b
Show file tree

Hide file tree

Showing 12 changed files with 3 additions and 3 deletions.
diff --git a/documentation/tensorflow_serving.md b/documentation/tensorflow_serving.md
@@ -39,7 +39,7 @@ In this example, we use an already trained TF-DF model.
 ```shell
 # Get a TF-DF model
 git clone https://github.com/tensorflow/decision-forests.git
-MODEL_PATH=$(pwd)/decision-forests/tensorflow_decision_forests/test_data/model/saved_model_adult_gbt
+MODEL_PATH=$(pwd)/decision-forests/tensorflow_decision_forests/test_data/model/saved_model_adult_rf
 
 echo "The TF-DF model is available at: ${MODEL_PATH}"
 ```

diff --git a/tensorflow_decision_forests/keras/keras_test.py b/tensorflow_decision_forests/keras/keras_test.py
@@ -2278,7 +2278,7 @@ def test_properties(self):
   def test_golden_model_gbt(self):
     dataset = adult_dataset()
     loaded_model = models.load_model(
-        os.path.join(tfdf_test_data_path(), "model/saved_model_adult_gbt")
+        os.path.join(tfdf_test_data_path(), "model/saved_model_adult_rf")
     )
     prediction = loaded_model.predict(
         keras.pd_dataframe_to_tf_dataset(dataset.test, label="income")

diff --git a/tensorflow_decision_forests/test_data/README.md b/tensorflow_decision_forests/test_data/README.md
@@ -3,5 +3,5 @@
 This directory only contains small files used for the unit tests.
 
 This directory structure is organized as follows: - model: Pre-trained models. -
-saved_model_adult_gbt: TF SavedModel containing a TF-DF GBT model trained with
+saved_model_adult_rf: TF SavedModel containing a TF-DF GBT model trained with
 Keras.
diff --git a/...saved_model_adult_gbt/assets/data_spec.pb → .../saved_model_adult_rf/assets/data_spec.pb b/...saved_model_adult_gbt/assets/data_spec.pb → .../saved_model_adult_rf/assets/data_spec.pb
diff --git a/...a/model/saved_model_adult_gbt/assets/done → ...ta/model/saved_model_adult_rf/assets/done b/...a/model/saved_model_adult_gbt/assets/done → ...ta/model/saved_model_adult_rf/assets/done
diff --git a/...el/saved_model_adult_gbt/assets/header.pb → ...del/saved_model_adult_rf/assets/header.pb b/...el/saved_model_adult_gbt/assets/header.pb → ...del/saved_model_adult_rf/assets/header.pb
diff --git a/...del_adult_gbt/assets/nodes-00000-of-00001 → ...odel_adult_rf/assets/nodes-00000-of-00001 b/...del_adult_gbt/assets/nodes-00000-of-00001 → ...odel_adult_rf/assets/nodes-00000-of-00001
diff --git a/..._adult_gbt/assets/random_forest_header.pb → ...l_adult_rf/assets/random_forest_header.pb b/..._adult_gbt/assets/random_forest_header.pb → ...l_adult_rf/assets/random_forest_header.pb
diff --git a/...l/saved_model_adult_gbt/keras_metadata.pb → ...el/saved_model_adult_rf/keras_metadata.pb b/...l/saved_model_adult_gbt/keras_metadata.pb → ...el/saved_model_adult_rf/keras_metadata.pb
diff --git a/...odel/saved_model_adult_gbt/saved_model.pb → ...model/saved_model_adult_rf/saved_model.pb b/...odel/saved_model_adult_gbt/saved_model.pb → ...model/saved_model_adult_rf/saved_model.pb
diff --git a/...t/variables/variables.data-00000-of-00001 → ...f/variables/variables.data-00000-of-00001 b/...t/variables/variables.data-00000-of-00001 → ...f/variables/variables.data-00000-of-00001
diff --git a/...model_adult_gbt/variables/variables.index → ..._model_adult_rf/variables/variables.index b/...model_adult_gbt/variables/variables.index → ..._model_adult_rf/variables/variables.index