我刚开始学习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)