skflow.TensorFlowDNNRegressor 参数

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

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注