Как напечатать обученные параметры классификатора в TensorFlow

Question

Я обучил модель в TensorFlow и сохранил ее на диске. Теперь я хочу загрузить его с контрольной точки и распечатать обученные параметры. Что-то вроде:

classifier = tf.estimator.DNNClassifier(
    feature_columns=feature_columns,
    hidden_units=hidden_units,
    warm_start_from=checkpoint_path)

print(parameters(classifier))

Как мне это сделать? Я использую TF версии 1.14.

Answer 1

Я думаю, вы можете использовать эти два метода get_variable_names() и get_variable_value() для получения параметров в вашем классификаторе.

params = classifier.get_variable_names()
for p in params:
    print(p, classifier.get_variable_value(p))