fix bug of autoint model demo config

alibaba · Dec 20, 2024 · 4686391 · 4686391
1 parent 27a8622
commit 4686391
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/easy_rec/python/inference/predictor.py b/easy_rec/python/inference/predictor.py
@@ -534,7 +534,7 @@ def _parse_value(all_vals):
               ]
           for k in self._reserved_cols:
             if k in all_vals and all_vals[k].dtype == np.object:
-              all_vals[k] = [val.decode('utf-8') for val in all_vals[k]]
+              all_vals[k] = [val.decode('utf-8', errors='ignore') for val in all_vals[k]]
 
           ts2 = time.time()
           reserve_vals = self._get_reserve_vals(self._reserved_cols,

diff --git a/examples/configs/autoint_on_movielens.config b/examples/configs/autoint_on_movielens.config
@@ -3,7 +3,7 @@ eval_input_path: "examples/data/movielens_1m/movies_test_data"
 model_dir: "examples/ckpt/autoint_on_movieslen_ckpt"
 
 train_config {
-  log_step_count_steps: 100
+  log_step_count_steps: 1000
   optimizer_config: {
     adam_optimizer: {
       learning_rate: {
@@ -157,5 +157,5 @@ model_config: {
   embedding_regularization: 1e-4
 }
 export_config {
-  multi_placeholder: false
+  multi_placeholder: true
 }