2016-05-19 1 views
0

私はskflowが初めてです。次のコード例では、神経回路網推定器を初期化することができます。skflow.TensorFlowDNNRegressorパラメータ

regressor = skflow.TensorFlowDNNRegressor(
    hidden_units=[10, 10], 
    steps=5000, 
    learning_rate=0.1, 
    batch_size=1) 
私はこれらがTensorFlowDNNRegressorのための唯一の入力パラメータは、以上のパラメータがある場合、私は変更することができ、知っていただきたいと思い

?誰でも私にこのパラメータのリストがどこにあるのか教えてください。私はそれについてのドキュメントを見つけることができません。

+0

ドキュメントがすぐに来ています。乞うご期待! –

+0

@ YuanTang更新いただきありがとうございます。ドキュメンテーションを楽しみにしています。 –

答えて

3

この機能のドキュメントは、hereです。 (それほどきれいなフォーマットではない)。ここで

は、クラスの異なるパラメータです:

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) 
+0

@Oliver Moindrot:このリンクをありがとう。 –