demo

Author: formath

README.md

@@ -21,27 +21,35 @@ cmake ..
 make
 ```
 
-# Demo
+# Simple Demo
+This demo used `c=a+b` to show how to save the model and load it using C++ for prediction. [tensorflow_c++_api_prediction_basic](http://mathmach.com/2017/10/09/tensorflow_c++_api_prediction_basic/)
+```bash
+cd demo/simple_model
+# train
+sh train.sh
+# predict
+sh predict.sh
+```
 
-## Transform text file into TFRecord
+# Deep CTR Model Demo
+This demo show a real-wrold deep model usage in click through rate prediction. [tensorflow_c++_api_prediction_advance](http://mathmach.com/2017/10/11/tensorflow_c++_api_prediction_advance/)
+
+## Transform LibFM data into TFRecord
+* LibFM format: `label fieldId:featureId:value ...`
 ```bash
-cd demo
+cd demo/deep_model
 sh trans_data_to_tfrecord.sh
 cd ..
 ```
 
 ## Train model
 ```bash
-cd demo
 sh train.sh
-cd ..
 ```
 
 ## Predict using C++
 ```bash
-cd demo
-sh test.sh
-cd ..
+sh predict.sh
 ```
 
 # Reference

demo/libfm.data renamed to demo/deep_model/data/libfm.data

@@ -11,6 +11,6 @@
 #    --output_node_names=cross_entropy
 #cd -
 
-python ../python/freeze_graph.py \
-    --model_dir=./model \
+python ../../python/freeze_graph.py \
+    --model_dir=./saved_model \
     --output_node_names=Softmax

demo/freeze_graph.sh renamed to demo/deep_model/freeze_graph.sh

@@ -0,0 +1,3 @@
+#!/usr/bin/env bash
+
+# TODO

demo/deep_model/predict.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+
+python ../../python/train.py \
+    --dict "./data/dict.data" \
+    --continuous_fields "" \
+    --sparse_fields "9,6,116" \
+    --linear_fields "152,179" \
+    --train_file "./data/libfm.tfrecord" \
+    --valid_file "./data/libfm.tfrecord"

demo/deep_model/train.sh

@@ -0,0 +1,18 @@
+#!/usr/bin/env bash
+
+# generate field dict
+python ../../python/dict.py \
+	'0' \
+	'9,6,116' \
+	'152,179' \
+	./data/libfm.data \
+	./data/dict.data
+
+# transform libfm data into tfrecord
+python ../../python/data.py \
+	./data/dict.data \
+	'0' \
+	'9,6,116' \
+	'152,179' \
+	./data/libfm.data \
+	./data/libfm.tfrecord

demo/deep_model/trans_data_to_tfrecord.sh

@@ -0,0 +1,3 @@
+#!/usr/bin/env bash
+
+../../bin/simple_model.bin "./saved_model/graph.pb"

demo/simple_model/predict.sh

@@ -0,0 +1,3 @@
+#!/usr/bin/env bash
+
+python ../../python/simple_model.py