structure rebuild

fleetrec/core/factory.py

@@ -17,12 +17,11 @@ import sys
 
 import yaml
 
-from fleetrec.trainer.local_engine import Launch
-from fleetrec.trainer.single_trainer import SingleTrainer
-from fleetrec.trainer.cluster_trainer import ClusterTrainer
-from fleetrec.trainer.ctr_trainer import CtrPaddleTrainer
+from fleetrec.core.trainers.single_trainer import SingleTrainer
+from fleetrec.core.trainers.cluster_trainer import ClusterTrainer
+from fleetrec.core.trainers.ctr_trainer import CtrPaddleTrainer
 
-from fleetrec.utils import envs
+from fleetrec.core.utils import envs
 
 
 class TrainerFactory(object):

fleetrec/core/layer.py

@@ -13,12 +13,7 @@
 # limitations under the License.
 
 import abc
-import copy
-import yaml
-import paddle.fluid as fluid
-from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet
 
-from fleetrec.core.utils import table as table
 
 class Layer(object):
     """R

setup.py

@@ -5,12 +5,12 @@ setup for fleet-rec.
 from setuptools import setup
 
 packages = ["fleetrec", "fleetrec.models",
-            "fleetrec.examples", "fleetrec.examples.user_define", "fleetrec.examples.build_in"
+            "fleetrec.examples", "fleetrec.examples.user_define", "fleetrec.examples.build_in",
             "fleetrec.core", "fleetrec.core.engine",
             "fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"]
 
 requires = [
-    "paddlepaddle"
+    "paddlepaddle >= 0.0.0"
 ]
 
 about = {}