import os
import numpy as np
import tensorflow as tf
from tensorflow import saved_model as sm
with tf.Session() as sess:
path = './serving_models/DeepFMPos/1'
sm.loader.load(sess, tags=[sm.tag_constants.SERVING], export_dir=path)
ret = [v for v in tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES) if v.name.endswith('pos_W:0')]
print(ret)
W = sess.run('DeepFMPos/deep_fm/pos_W:0')
for i in range(0, 100):
print('pos:{},weight:{}'.format(i,W[i]))
从Tensorflow训练好的模型中获取参数（权重）
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
- 文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
推荐阅读更多精彩内容
- 【蝴蝶效应】 蝴蝶效应:上个世纪70年代，美国一个名叫洛伦兹的气象学家在解释空气系统理论时说，亚马逊雨林一只蝴蝶...
- from tensorflow.python.toolsimport inspect_checkpointas c...
