Как извлечь весовые коэффициенты из смещения нейросети тензорного потока и как самостоятельно рассчитать прогноз в python?

Question

Я использую tenorflow в python для построения простой нейронной сети для регрессии, и я хотел бы извлечь весовые коэффициенты и смещение, чтобы использовать их в другой программе.Может кто-нибудь может указать, как извлечь весовые коэффициенты из нейронной сети и вычислить прогноз на основе тех же весовых коэффициентов, с простым умножением и сложением матриц после обучения NN тензорного потока.Также это вычисление улучшит время предсказания (так как tf.predict делает некоторые дополнительные вычисления)?

Answer 1

Вы можете использовать класс tf.train.Saver, чтобы сохранить модель в виде файла ckpt и использовать его позже:

saver = tf.train.Saver()
 with tf.Session() as sess:
    #your code here
    save_path = saver.save(sess, "/model_path/model.ckpt")

Чтобы восстановить переменные:

saver.restore(sess, "/your_path/model.ckpt") 

Полная документация: https://www.tensorflow.org/guide/saved_model