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

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

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