我是skflow的新手。通过下面的示例代码,我能够初始化一个神经网络估计器。
regressor = skflow.TensorFlowDNNRegressor(
hidden_units=[10, 10],
steps=5000,
learning_rate=0.1,
batch_size=1)
我想知道这些是TensorFlowDNNRegressor的唯一输入参数,还是有更多的参数,我可以更改?有人能告诉我在哪里可以找到这个参数列表吗?我找不到任何文档。
我在这里找到了这个函数的文档。(格式还不是很干净)。
以下是该类的不同参数:
skflow.TensorFlowDNNRegressor.__init__(
hidden_units,
n_classes=0,
tf_master='',
batch_size=32,
steps=200,
optimizer='SGD',
learning_rate=0.1,
tf_random_seed=42,
continue_training=False,
config_addon=None,
verbose=1,
max_to_keep=5,
keep_checkpoint_every_n_hours=10000)