张量流中Logistic回归模型的系数

Question

我跟随此tutorial使用TensorFlow构建逻辑回归模型。我有一个简单的问题：如何在.coef_中打印模型的系数，如sklearn？

预处理之后和训练模型之前的一段代码：

model_dir = tempfile.mkdtemp()
m = tf.contrib.learn.LinearClassifier(feature_columns=[
  gender, native_country, education, occupation, workclass, marital_status, race,
  age_buckets, education_x_occupation, age_buckets_x_education_x_occupation],
  model_dir=model_dir)
m.fit(input_fn=train_input_fn, steps=200)
results = m.evaluate(input_fn=eval_input_fn, steps=1)
for key in sorted(results):
    print("%s: %s" % (key, results[key]))

以下是ls -l上model_dir的结果：

-rw-r--r--  1 builder  staff       305 Jul 17 13:29 checkpoint
drwxr-xr-x  3 builder  staff       102 Jul 17 13:29 eval
-rw-r--r--  1 builder  staff  39038807 Jul 17 13:29 events.out.tfevents.1500323287.builder-MBP.ikuni.com
-rw-r--r--  1 builder  staff  32604586 Jul 17 13:28 graph.pbtxt
-rw-r--r--  1 builder  staff  12567520 Jul 17 13:28 model.ckpt-1.data-00000-of-00001
-rw-r--r--  1 builder  staff      3240 Jul 17 13:28 model.ckpt-1.index
-rw-r--r--  1 builder  staff   9830014 Jul 17 13:28 model.ckpt-1.meta
-rw-r--r--  1 builder  staff  12567520 Jul 17 13:29 model.ckpt-200.data-00000-of-00001
-rw-r--r--  1 builder  staff      3240 Jul 17 13:29 model.ckpt-200.index
-rw-r--r--  1 builder  staff   9830014 Jul 17 13:29 model.ckpt-200.meta

Answer 1

您可以使用两种方法：get_variable_names()来获取模型中的变量，然后使用get_variable_value()来获取训练模型的特定变量的系数。