diff --git a/fleetrec/core/models/modul/__init__.py b/fleetrec/core/models/modul/__init__.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/fleetrec/contrib/__init__.py b/fleetrec/core/modules/__init__.py similarity index 100% rename from fleetrec/contrib/__init__.py rename to fleetrec/core/modules/__init__.py diff --git a/fleetrec/core/models/__init__.py b/fleetrec/core/modules/coding/__init__.py similarity index 100% rename from fleetrec/core/models/__init__.py rename to fleetrec/core/modules/coding/__init__.py diff --git a/fleetrec/core/models/coding/layers.py b/fleetrec/core/modules/coding/layers.py similarity index 100% rename from fleetrec/core/models/coding/layers.py rename to fleetrec/core/modules/coding/layers.py diff --git a/fleetrec/core/models/coding/__init__.py b/fleetrec/core/modules/modul/__init__.py similarity index 100% rename from fleetrec/core/models/coding/__init__.py rename to fleetrec/core/modules/modul/__init__.py diff --git a/fleetrec/core/models/modul/build.py b/fleetrec/core/modules/modul/build.py similarity index 100% rename from fleetrec/core/models/modul/build.py rename to fleetrec/core/modules/modul/build.py diff --git a/fleetrec/core/models/modul/layers.py b/fleetrec/core/modules/modul/layers.py similarity index 100% rename from fleetrec/core/models/modul/layers.py rename to fleetrec/core/modules/modul/layers.py diff --git a/fleetrec/core/trainers/ctr_modul_trainer.py b/fleetrec/core/trainers/ctr_modul_trainer.py index bc923ff8ea68f85bae56057f71be4d44d072d16a..e72715e74634791c2c90890bc6b8b8f0a0e56c71 100755 --- a/fleetrec/core/trainers/ctr_modul_trainer.py +++ b/fleetrec/core/trainers/ctr_modul_trainer.py @@ -27,7 +27,7 @@ from paddle.fluid.incubate.fleet.base.role_maker import GeneralRoleMaker from fleetrec.core.utils import fs as fs from fleetrec.core.utils import util as util from fleetrec.core.metrics.auc_metrics import AUCMetric -from fleetrec.core.models.modul import build as model_basic +from fleetrec.core.modules.modul import build as model_basic from fleetrec.core.utils import dataset from fleetrec.core.trainer import Trainer diff --git a/setup.py b/setup.py index 1a67b939d6ebbfad6c778b3af8af24ed297e9d57..977b9915cbc22042a35f7dda9915f88efb7604e3 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ models_recall = ["fleetrec.models.recall", "fleetrec.models.recall.word2vec"] models = ["fleetrec.models"] + models_ctr + models_ot + models_recall core = ["fleetrec.core", "fleetrec.core.engine", - "fleetrec.core.metrics", "fleetrec.core.models", + "fleetrec.core.metrics", "fleetrec.core.modules", "fleetrec.core.trainers", "fleetrec.core.utils"] packages = ["fleetrec", "fleetrec.contrib", "fleetrec.dataset", "fleetrec.doc", "fleetrec.examples",