diff --git a/demo/ctr-dnn_train.yaml b/demo/ctr-dnn_train.yaml index ffa3de2b143a5adff977ebccf8f07f52675ae356..1303211bad4ff5f2db2c9aeab342d6642de21d17 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.ranK.criteo_reader" - train_data_path: "fleetrec/models/ranK/dnn/data/train" + class: "fleetrec.models.rank.criteo_reader" + train_data_path: "fleetrec/models/rank/dnn/data/train" model: - models: "fleetrec.models.ranK.dnn.model" + models: "fleetrec.models.rank.dnn.model" hyper_parameters: sparse_inputs_slots: 27 sparse_feature_number: 1000001 diff --git a/models/rank/dnn/hyper_parameters.yaml b/models/rank/dnn/hyper_parameters.yaml deleted file mode 100644 index 85048fa37155751c9542322525f15fa1fe3870fd..0000000000000000000000000000000000000000 --- a/models/rank/dnn/hyper_parameters.yaml +++ /dev/null @@ -1,20 +0,0 @@ -# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -sparse_inputs_slots: 27, -sparse_feature_number: 1000001, -sparse_feature_dim: 8, -dense_input_dim: 13, -fc_sizes: [400, 400, 40], -learning_rate: 0.001