如何在TensorFlow中

时间:2017-08-03 13:38:57

标签: python tensorflow

我想在TensorFlow中为相同配置的深度神经网络模型应用不同的激活函数。这就是为什么这些分类器的初始权重应该相同。但是,我无法找到解决此任务的方法。我应该将first_classifier克隆到second_classifier,还是将first_classifier的权重转移到第二个分类器。感谢您的支持。

first_classifier = tf.contrib.learn.DNNClassifier(
    feature_columns = feature_columns
    , hidden_units = [3]
    , activation_fn = tf.nn.sigmoid
    , optimizer = tf.train.GradientDescentOptimizer(learningRate))

second_classifier = tf.contrib.learn.DNNClassifier(
    feature_columns = feature_columns
    , hidden_units = [3]
    , activation_fn = tf.nn.tanh
    , optimizer = tf.train.GradientDescentOptimizer(learningRate))

1 个答案:

答案 0 :(得分:0)

DNN docs开始,您无法直接设置初始权重;除了使用set_params。最好的方法是在两个分类器之间共享参数(克隆或传输)。