python - 从 TensorFlow 的 estimator.DNNClassifier 获取权重

Question

我知道 DNNClassifier 现在是通过 estimator.DNNClassifier 训练的。在使用 contrib.learn.DNNClassifier 对其进行训练之前，我们可以使用 get_variable_names() 提取权重。但是estimator.DNNClassifier中没有这样的方法。如果现在不推荐使用 contrib.learn，那么我们如何从新的 estimator.DNNClassifier 中获取权重？

Answer 1

显然，权重被称为“内核”(learnt from this question)

例如，对于:

estimator = tf.estimator.DNNClassifier(
   feature_columns=feature_columns, 
   hidden_units=[2])

estimator.train(input_fn=input_fn_train)

您可以像这样使用 get_variable_value:

print(estimator.get_variable_value("dnn/hiddenlayer_0/kernel"))
print(estimator.get_variable_value("dnn/hiddenlayer_0/bias"))