#edit-mode: -*- python -*- # Copyright (c) 2016 Baidu, Inc. 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. import sys sys.path.append("..") from seqToseq_net import * # whether this config is used for generating is_generating = False ### Data Definiation data_dir = "./data/pre-wmt14" train_conf = seq_to_seq_data(data_dir = data_dir, is_generating = is_generating) ### Algorithm Configuration settings( learning_method = AdamOptimizer(), batch_size = 50, learning_rate = 5e-4) ### Network Architecture gru_encoder_decoder(train_conf, is_generating)