2017-07-18 96 views
1

我按照这个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])) 

这里是ls -lmodel_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 

回答

1

您可以使用两种方法:get_variable_names()得到变量,然后使用get_variable_value()来获得训练模型的特定变量的系数。