add internal data

demo/ctr-dnn_train.yaml

@@ -22,7 +22,7 @@ train:
   reader:
     batch_size: 2
     class: "fleetrec.models.rank.criteo_reader"
-    train_data_path: "models/rank/dnn/data/train"
+    train_data_path: "fleetrec::models/rank/dnn/data/train"
 
   model:
     models: "fleetrec.models.rank.dnn.model"

fleet_rec/run.py

@@ -183,6 +183,9 @@ if __name__ == "__main__":
         raise IOError("argument model: {} do not exist".format(args.model))
     engine_registry()
 
+    abs_dir = os.path.dirname(os.path.abspath(__file__))
+    envs.set_runtime_environs({"PACKAGE_BASE": abs_dir})
+
     which_engine = get_engine(args.engine, args.device)
 
     engine = which_engine(args)

setup.py

@@ -50,6 +50,8 @@ def build(dirname):
     packages = find_packages(dirname, include=('fleetrec.*'))
     package_dir = {'': dirname}
 
+    package_data = {'fleetrec.models.rank.dnn': ['data/*/*.txt'], }
+
     setup(
         name=about["__title__"],
         version=about["__version__"],
@@ -60,6 +62,7 @@ def build(dirname):
         url=about["__url__"],
         packages=packages,
         package_dir=package_dir,
+        package_data=package_data,
         python_requires=">=2.7",
         install_requires=requires,
         zip_safe=False