使用TensorFlow进行在线核外Logistic回归

时间:2017-05-31 17:44:30

标签: python machine-learning tensorflow classification large-data

我目前正在尝试使用TensorFlow(特别是tf.contrib.learn.LinearClassifier估算器)来训练在线逻辑回归:

optimizer = tf.train.AdagradOptimizer(learning_rate=1.0,initial_accumulator_value=0.4)
m = tf.contrib.learn.LinearClassifier(model_dir=model_dir,
                                      feature_columns=wide_columns,
                                      optimizer=optimizer)


m.partial_fit(input_fn=lambda: input_fn(df_train),batch_size=1)

不幸的是我收到以下错误:

ValueError: Can not provide both input_fn and batch_size.

看起来不推荐使用batch_size参数。在这种情况下,是否有人知道进行在线学习的简单方法?

由于

1 个答案:

答案 0 :(得分:1)

如文件

所述
  

参数x,y和batch_size仅在SKCompat类中可用,Estimator仅接受input_fn。示例转换:est = Estimator(...) - > est = SKCompat(Estimator(...))

尝试使用tf.contrib.learn.SKCompat