python - tensorflow 中逻辑回归模型的系数

Question

我正在关注这个tutorial使用 TensorFlow 构建逻辑回归模型。我有一个简单的问题:如何在 sklearn 中打印模型的系数，如 .coef_？

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

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]))

这是 model_dir 上 ls -l 的结果:

-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() 获取模型中特定变量的系数训练有素的模型。