second ci

Author: formath

CMakeLists.txt

@@ -0,0 +1,11 @@
+cmake_minimum_required(VERSION 2.8)
+
+project(deep-ctr)
+
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -fopenmp -msse2 -fPIC -O3 -ggdb -Wall -finline-functions")
+
+message(STATUS "CMAKE_CXX_COMPILER: ${CMAKE_CXX_COMPILER}")
+message(STATUS "CMAKE_CXX_FLAGS: ${CMAKE_CXX_FLAGS}")
+
+# Add Subdirectory
+add_subdirectory(src)

README.md

@@ -1,2 +1,51 @@
 # deep-ctr
 deep models for click through rate prediction
+
+# Build
+
+## Build TensorFlow
+Follow the instruction [build tensorflow from source](https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/makefile)
+```bash
+git clone --recursive https://github.com/tensorflow/tensorflow.git
+cd tensorflow
+sh tensorflow/contrib/makefile/build_all_ios.sh (depends on your platform)
+cd ..
+```
+
+## Build deep-ctr
+```bash
+git clone https://github.com/formath/deep-ctr.git
+cd deep-ctr
+mkdir build && cd build
+cmake ..
+make
+```
+
+# Demo
+
+## Transform text file into TFRecord
+```bash
+cd demo
+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 ..
+```
+
+# Reference
+* [Loading a TensorFlow graph with the C++ API](https://medium.com/jim-fleming/loading-a-tensorflow-graph-with-the-c-api-4caaff88463f)
+* [Loading a tensorflow graph with the C++ API by using Mnist](http://jackytung8085.blogspot.jp/2016/06/loading-tensorflow-graph-with-c-api-by.html)
+* [Tensorflow Different ways to Export and Run graph in C++](https://stackoverflow.com/questions/35508866/tensorflow-different-ways-to-export-and-run-graph-in-c/43639305#43639305)
+

data.py

@@ -0,0 +1,15 @@
+import tensorflow as tf
+import numpy as np
+
+with tf.Session() as sess:
+    a = tf.Variable(5.0, name='a')
+    b = tf.Variable(6.0, name='b')
+    c = tf.multiply(a, b, name="c")
+
+    sess.run(tf.global_variables_initializer())
+
+    print(a.eval()) # 5.0
+    print(b.eval()) # 6.0
+    print(c.eval()) # 30.0
+    
+    tf.train.write_graph(sess.graph_def, 'model', 'graph.pb', as_text=False)

test/libfm.data demo/libfm.data

@@ -1,6 +1,6 @@
 #!/usr/bin/env bash
 
-python ../train.py \
+python ../python/train.py \
     --dict dict.data \
     --continuous_fields "" \
     --sparse_fields "9,6,116" \

demo/test.py

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