Fix example2 data format to npz (#82)

reczoo · Feb 25, 2024 · e4d6045 · e4d6045
1 parent df04ae1
commit e4d6045
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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