GitHubじゃ!Pythonじゃ!

GitHubからPython関係の優良リポジトリを探したかったのじゃー、でも英語は出来ないから日本語で読むのじゃー、英語社会世知辛いのじゃー

deepmind

sonnet – TensorFlowベースのニューラルネットワークライブラリ

投稿日:

TensorFlowベースのニューラルネットワークライブラリ

Sonnetは、複雑なニューラルネットワークを構築するためのTensorFlowの上に構築されたライブラリです。

インストール

Sonnetは、GPUをサポートしているかどうかにかかわらずpipからインストールできます。

このインストールは、Linux / Mac OS XおよびPython 2.7および3. {4,5,6}と互換性があります。 インストールされるTensorFlowのバージョンは1.5でなければなりません。 Sonnetをインストールすると、 TensorFlowのvirtualenvインストールモードネイティブpipインストールがサポートされます。

sonnetをインストールするには、次のコマンドを実行します。

$ pip install dm-sonnet

Sonnetは、テンソルフローのCPUとGPUバージョンの両方で動作しますが、Tensorflowを要件としてリストしていないため、Tensorflowを別途インストールする必要があります。

使用例

次のコードは、Linearモジュールを構築し、それを複数の入力に接続します。 変数(すなわち、線形変換の重みと偏り)は自動的に共有されます。

import sonnet as snt

# Provide your own functions to generate data Tensors.
train_data = get_training_data()
test_data = get_test_data()

# Construct the module, providing any configuration necessary.
linear_regression_module = snt.Linear(output_size=FLAGS.output_size)

# Connect the module to some inputs, any number of times.
train_predictions = linear_regression_module(train_data)
test_predictions = linear_regression_module(test_data)

ドキュメンテーション

ここの完全なドキュメントページを調べてください。







-deepmind

執筆者: