# 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.fromsentiment_netimport*frompaddle.trainer_config_helpersimport*# whether this config is used for testis_test=get_config_arg('is_test',bool,False)# whether this config is used for predictionis_predict=get_config_arg('is_predict',bool,False)data_dir="./data/pre-imdb"dict_dim,class_dim=sentiment_data(data_dir,is_test,is_predict)################## Algorithm Config #####################settings(batch_size=128,learning_rate=2e-3,learning_method=AdamOptimizer(),regularization=L2Regularization(8e-4),gradient_clipping_threshold=25)#################### Network Config ######################stacked_lstm_net(dict_dim,class_dim=class_dim,stacked_num=3,is_predict=is_predict)# bidirectional_lstm_net(dict_dim, class_dim=class_dim, is_predict=is_predict)
