diff --git a/demo/ctr-dnn_train.yaml b/demo/ctr-dnn_train.yaml index 4d2644e366ed638136fad3767b4035e5ac950c60..ffa3de2b143a5adff977ebccf8f07f52675ae356 100644 --- a/demo/ctr-dnn_train.yaml +++ b/demo/ctr-dnn_train.yaml @@ -29,11 +29,11 @@ train: reader: mode: "dataset" batch_size: 2 - class: "fleetrec.models.ctr.criteo_reader" - train_data_path: "fleetrec/models/ctr/dnn/data/train" + class: "fleetrec.models.ranK.criteo_reader" + train_data_path: "fleetrec/models/ranK/dnn/data/train" model: - models: "fleetrec.models.ctr.dnn.model" + models: "fleetrec.models.ranK.dnn.model" hyper_parameters: sparse_inputs_slots: 27 sparse_feature_number: 1000001 diff --git a/models/ctr/deepfm/__init__.py b/models/ctr/deepfm/__init__.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/multi_task/__init__.py b/models/multi_task/__init__.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/other/__init__.py b/models/other/__init__.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/ctr/__init__.py b/models/rank/__init__.py similarity index 100% rename from models/ctr/__init__.py rename to models/rank/__init__.py diff --git a/models/ctr/criteo_reader.py b/models/rank/criteo_reader.py similarity index 100% rename from models/ctr/criteo_reader.py rename to models/rank/criteo_reader.py diff --git a/models/ctr/dnn/__init__.py b/models/rank/dnn/__init__.py similarity index 100% rename from models/ctr/dnn/__init__.py rename to models/rank/dnn/__init__.py diff --git a/models/ctr/dnn/data/test/sample_test.txt b/models/rank/dnn/data/test/sample_test.txt similarity index 100% rename from models/ctr/dnn/data/test/sample_test.txt rename to models/rank/dnn/data/test/sample_test.txt diff --git a/models/ctr/dnn/data/train/sample_train.txt b/models/rank/dnn/data/train/sample_train.txt similarity index 100% rename from models/ctr/dnn/data/train/sample_train.txt rename to models/rank/dnn/data/train/sample_train.txt diff --git a/models/ctr/dnn/hyper_parameters.yaml b/models/rank/dnn/hyper_parameters.yaml similarity index 100% rename from models/ctr/dnn/hyper_parameters.yaml rename to models/rank/dnn/hyper_parameters.yaml diff --git a/models/ctr/dnn/model.py b/models/rank/dnn/model.py similarity index 100% rename from models/ctr/dnn/model.py rename to models/rank/dnn/model.py diff --git a/models/recall/__init__.py b/models/recall/__init__.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/recall/word2vec/hyper_parameters.yaml b/models/recall/word2vec/hyper_parameters.yaml deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/recall/word2vec/model.py b/models/recall/word2vec/model.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/recall/word2vec/reader.py b/models/recall/word2vec/reader.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/recall/word2vec/sample_test.txt b/models/recall/word2vec/sample_test.txt deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/recall/word2vec/sample_train.txt b/models/recall/word2vec/sample_train.txt deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/models/tdm/__init__.py b/models/tdm/__init__.py deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000