From 4686391a6ce004eef529536186c038888d1c8a0f Mon Sep 17 00:00:00 2001 From: yangxudong Date: Fri, 20 Dec 2024 14:52:39 +0800 Subject: [PATCH] fix bug of autoint model demo config --- easy_rec/python/inference/predictor.py | 2 +- examples/configs/autoint_on_movielens.config | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/easy_rec/python/inference/predictor.py b/easy_rec/python/inference/predictor.py index 93220f047..fa0853ea5 100644 --- 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 index f74918899..35f0009ae 100644 --- 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 }