fix

Author: formath

Diff for: demo/deep_model/predict.sh

@@ -1,7 +1,6 @@
 #!/usr/bin/env bash
 
 ../../bin/deep_model.bin \
-	"9,6,116" \
-    "152,179" \
-    "" \
+	"9,6,116,152,179" \
+    "./data/dict.data" \
 	"./model/freeze_graph.pb"

Diff for: demo/deep_model/train.sh

@@ -4,9 +4,7 @@
 # to save model and checkpoint
 python ../../python/train.py \
     --dict "./data/dict.data" \
-    --continuous_fields "" \
-    --sparse_fields "9,6,116" \
-    --linear_fields "152,179" \
+    --sparse_fields "9,6,116,152,179" \
     --train_file "./data/libfm.tfrecord" \
     --valid_file "./data/libfm.tfrecord"
 
@@ -15,8 +13,6 @@ python ../../python/train.py \
 # for feed Tensor when prediction
 python ../../python/predict_model.py \
     --dict "./data/dict.data" \
-    --continuous_fields "" \
-    --sparse_fields "9,6,116" \
-    --linear_fields "152,179" \
+    --sparse_fields "9,6,116,152,179" \
     --train_file "./data/libfm.tfrecord" \
     --valid_file "./data/libfm.tfrecord"

Diff for: demo/deep_model/trans_data_to_tfrecord.sh

@@ -2,9 +2,7 @@
 
 # generate field dict
 python ../../python/dict.py \
-	'0' \
-	'9,6,116' \
-	'152,179' \
+	'9,6,116,152,179' \
 	./data/libfm.data \
 	./data/dict.data
 
@@ -15,9 +13,7 @@ fi
 # transform libfm data into tfrecord
 python ../../python/data.py \
 	./data/dict.data \
-	'0' \
-	'9,6,116' \
-	'152,179' \
+	'9,6,116,152,179' \
 	./data/libfm.data \
 	./data/libfm.tfrecord