From e4d604543f45d70f5be97f69940ad2b29109521c Mon Sep 17 00:00:00 2001 From: lu-minous <48793299+lu-minous@users.noreply.github.com> Date: Sun, 25 Feb 2024 08:11:30 +0800 Subject: [PATCH] Fix example2 data format to npz (#82) --- data/tiny_npz/feature_map.json | 2 +- demo/config/example2_config/model_config.yaml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/data/tiny_npz/feature_map.json b/data/tiny_npz/feature_map.json index cc914f7..e1d6e02 100644 --- a/data/tiny_npz/feature_map.json +++ b/data/tiny_npz/feature_map.json @@ -1,5 +1,5 @@ { - "dataset_id": "tiny_h5", + "dataset_id": "tiny_npz", "num_fields": 14, "total_features": 485, "input_length": 14, diff --git a/demo/config/example2_config/model_config.yaml b/demo/config/example2_config/model_config.yaml index d0d3e6c..e7b4130 100644 --- a/demo/config/example2_config/model_config.yaml +++ b/demo/config/example2_config/model_config.yaml @@ -12,9 +12,9 @@ Base: feature_specs: null feature_config: null -DeepFM_test_h5: +DeepFM_test_npz: model: DeepFM - dataset_id: tiny_h5 + dataset_id: tiny_npz loss: 'binary_crossentropy' metrics: ['logloss', 'AUC'] task: binary_classification