保存されたモデルバンドル

パブリック クラスSavedModelBundle

SavedModelBundle は、ストレージからロードされたモデルを表します。

モデルは、計算の説明 ( Graph )、ストレージに保存された値に初期化されたテンソル (グラフ内のパラメーターや変数など) を含むSession 、およびモデルの説明 ( MetaGraphDef プロトコル バッファーのシリアル化された表現) で構成されます。 )。

ネストされたクラス

クラスSavedModelBundle.Loader SavedModel をロードするためのオプション。

パブリックメソッド

空所
近い()
保存されたモデル バンドルに関連付けられたリソース ( GraphおよびSession ) を解放します。
グラフ
グラフ()
モデルによって実行された計算を説明するグラフを返します。
静的SavedModelBundle
ロード(文字列エクスポートディレクトリ、文字列... タグ)
保存されたモデルをエクスポート ディレクトリからロードします。
静的SavedModelBundle.Loader
ローダー(文字列のエクスポートディレクトリ)
保存したモデルをロードします。
バイト[]
メタグラフ定義()
保存されたモデルに関連付けられたシリアル化されたMetaGraphDef プロトコル バッファを返します。
セッション
セッション()
モデルを使用して計算を実行するSessionを返します。

継承されたメソッド

パブリックメソッド

public void close ()

保存されたモデル バンドルに関連付けられたリソース ( GraphおよびSession ) を解放します。

公開グラフグラフ()

モデルによって実行された計算を説明するグラフを返します。

public static SavedModelBundleロード(String exportDir、String... タグ)

保存されたモデルをエクスポート ディレクトリからロードします。ロードされるモデルは、保存されたモデル APIを使用して作成する必要があります。

このメソッドは

SavedModelBundle.loader().withTags(tags).load();
 
の短縮形です。

パラメーター
エクスポートディレクトリ保存されたモデルを含むディレクトリ パス。
タグロードする特定のメタグラフ定義を識別するタグ。
戻り値
  • グラフと関連セッションを含むバンドル。

public static SavedModelBundle.Loaderローダー(String exportDir)

保存したモデルをロードします。

実際にモデルをロードする前に構成オプションを設定できるLoaderオブジェクトを返します。

パラメーター
エクスポートディレクトリ保存されたモデルを含むディレクトリ パス。

public byte[] metaGraphDef ()

保存されたモデルに関連付けられたシリアル化されたMetaGraphDef プロトコル バッファを返します。

公開セッションセッション()

モデルを使用して計算を実行するSessionを返します。

戻り値
  • 初期化されたセッション