Skip to content

Commit d965fb7

Browse files
authored
fix bug of autoint model demo config (#507)
1 parent 27a8622 commit d965fb7

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

easy_rec/python/inference/predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -534,7 +534,7 @@ def _parse_value(all_vals):
534534
]
535535
for k in self._reserved_cols:
536536
if k in all_vals and all_vals[k].dtype == np.object:
537-
all_vals[k] = [val.decode('utf-8') for val in all_vals[k]]
537+
all_vals[k] = [val.decode('utf-8', errors='ignore') for val in all_vals[k]]
538538

539539
ts2 = time.time()
540540
reserve_vals = self._get_reserve_vals(self._reserved_cols,

examples/configs/autoint_on_movielens.config

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ eval_input_path: "examples/data/movielens_1m/movies_test_data"
33
model_dir: "examples/ckpt/autoint_on_movieslen_ckpt"
44

55
train_config {
6-
log_step_count_steps: 100
6+
log_step_count_steps: 1000
77
optimizer_config: {
88
adam_optimizer: {
99
learning_rate: {
@@ -157,5 +157,5 @@ model_config: {
157157
embedding_regularization: 1e-4
158158
}
159159
export_config {
160-
multi_placeholder: false
160+
multi_placeholder: true
161161
}

0 commit comments

Comments
 (0)