update to fleetrec

examples/ctr-dnn_train_cluster.yaml → fleetrec/examples/ctr-dnn_train_cluster.yaml

@@ -41,11 +41,11 @@ train:
   reader:
     mode: "dataset"
     batch_size: 2
-    pipe_command: "python /paddle/fleet_rec/models/ctr_dnn/dataset.py"
-    train_data_path: "/paddle/fleet_rec/models/ctr_dnn/data/train"
+    pipe_command: "python /paddle/fleetrec/models/ctr_dnn/dataset.py"
+    train_data_path: "/paddle/fleetrec/models/ctr_dnn/data/train"
 
   model:
-    models: "fleet_rec.models.ctr_dnn.model"
+    models: "fleetrec.models.ctr_dnn.model"
     hyper_parameters:
       sparse_inputs_slots: 27
       sparse_feature_number: 1000001

examples/ctr-dnn_train_single.yaml → fleetrec/examples/ctr-dnn_train_single.yaml

@@ -35,11 +35,11 @@ train:
   reader:
     mode: "dataset"
     batch_size: 2 
-    pipe_command: "python /paddle/fleet_rec/models/ctr_dnn/dataset.py"
-    train_data_path: "/paddle/fleet_rec/models/ctr_dnn/data/train"
+    pipe_command: "python /paddle/fleetrec/models/ctr_dnn/dataset.py"
+    train_data_path: "/paddle/fleetrec/models/ctr_dnn/data/train"
 
   model:
-    models: "fleet_rec.models.ctr_dnn.model"
+    models: "fleetrec.models.ctr_dnn.model"
     hyper_parameters:
       sparse_inputs_slots: 27
       sparse_feature_number: 1000001

examples/train.py → fleetrec/examples/train.py

@@ -26,7 +26,7 @@
 
 import os
 
-from fleet_rec.trainer.factory import TrainerFactory
+from fleetrec.trainer.factory import TrainerFactory
 
 if __name__ == "__main__":
 

models/ctr_dnn/model.py → fleetrec/models/ctr_dnn/model.py

@@ -15,7 +15,7 @@
 import math
 import paddle.fluid as fluid
 
-from fleet_rec.utils import envs
+from fleetrec.utils import envs
 
 
 class Train(object):

trainer/ctr_trainer.py → fleetrec/trainer/ctr_trainer.py

@@ -23,11 +23,11 @@ import paddle.fluid as fluid
 from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet
 from paddle.fluid.incubate.fleet.base.role_maker import GeneralRoleMaker
 
-from fleet_rec.utils import fs as fs
-from fleet_rec.utils import util as util
-from fleet_rec.metrics.auc_metrics import AUCMetric
-from fleet_rec.models import base as model_basic
-from fleet_rec.reader import dataset
+from fleetrec.utils import fs as fs
+from fleetrec.utils import util as util
+from fleetrec.metrics.auc_metrics import AUCMetric
+from fleetrec.models import base as model_basic
+from fleetrec.reader import dataset
 from .trainer import Trainer
 
 

trainer/factory.py → fleetrec/trainer/factory.py

@@ -29,16 +29,16 @@ import sys
 
 import yaml
 
-from fleet_rec.trainer.single_trainer import SingleTrainerWithDataloader
-from fleet_rec.trainer.single_trainer import SingleTrainerWithDataset
+from fleetrec.trainer.single_trainer import SingleTrainerWithDataloader
+from fleetrec.trainer.single_trainer import SingleTrainerWithDataset
 
-from fleet_rec.trainer.cluster_trainer import ClusterTrainerWithDataloader
-from fleet_rec.trainer.cluster_trainer import ClusterTrainerWithDataset
+from fleetrec.trainer.cluster_trainer import ClusterTrainerWithDataloader
+from fleetrec.trainer.cluster_trainer import ClusterTrainerWithDataset
 
-from fleet_rec.trainer.local_engine import Launch
-from fleet_rec.trainer.ctr_trainer import CtrPaddleTrainer
+from fleetrec.trainer.local_engine import Launch
+from fleetrec.trainer.ctr_trainer import CtrPaddleTrainer
 
-from fleet_rec.utils import envs
+from fleetrec.utils import envs
 
 
 def str2bool(v):
@@ -103,7 +103,7 @@ class TrainerFactory(object):
             with open(config, 'r') as rb:
                 _config = yaml.load(rb.read(), Loader=yaml.FullLoader)
         else:
-            raise ValueError("fleet_rec's config only support yaml")
+            raise ValueError("fleetrec's config only support yaml")
 
         envs.set_global_envs(_config)
         mode = envs.get_global_env("train.trainer")

trainer/local_engine.py → fleetrec/trainer/local_engine.py

@@ -38,7 +38,7 @@ def start_procs(args, yaml):
     user_endpoints_ips = [x.split(":")[0] for x in user_endpoints.split(",")]
     user_endpoints_port = [x.split(":")[1] for x in user_endpoints.split(",")]
 
-    factory = "fleet_rec.trainer.factory"
+    factory = "fleetrec.trainer.factory"
     cmd = [sys.executable, "-u", "-m", factory, yaml]
 
     for i in range(server_num):

utils/envs.py → fleetrec/utils/envs.py

@@ -70,7 +70,7 @@ def pretty_print_envs(envs, header=None):
     if header:
         draws += h_format.format(header[0], header[1])
     else:
-        draws += h_format.format("fleet_rec Global Envs", "Value")
+        draws += h_format.format("fleetrec Global Envs", "Value")
 
     draws += line + "\n"