SavedModelBundle

パブリック クラスSavedModelBundle

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

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

ネストされたクラス

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

定数

弦DEFAULT_TAG

パブリックメソッド

Map<String, Tensor >
call (Map<String, Tensor > 引数)
このモデルからデフォルト関数を直接呼び出します。
空所
近い（）
保存されたモデル バンドルに関連付けられたリソース ( GraphおよびSession ) を解放します。
静的SavedModelBundle.Exporter
エクスポーター(文字列exportDir)
保存したモデルをエクスポートします。
具体的な関数
関数(文字列の署名キー)
関数シグネチャに対応するConcreteFunctionを返します。
グラフ
グラフ（）
モデルによって実行された計算を説明するグラフを返します。
静的SavedModelBundle
ロード(文字列エクスポートディレクトリ、文字列... タグ)
保存されたモデルをエクスポート ディレクトリからロードします。
静的SavedModelBundle.Loader
ローダー(文字列のエクスポートディレクトリ)
保存したモデルをロードします。
メタグラフ定義
メタグラフ定義()
保存されたモデルに関連付けられたMetaGraphDef プロトコル バッファを返します。
セッション
セッション（）
モデルを使用して計算を実行するSessionを返します。
リスト<署名>
署名()
この保存されたモデルで使用可能なすべての関数のシグネチャを返します。

継承されたメソッド

定数

パブリック静的最終文字列DEFAULT_TAG

定数値: 「サーブ」

パブリックメソッド

public Map<String, Tensor > call (Map<String, Tensor > 引数)

このモデルからデフォルト関数を直接呼び出します。

デフォルトの関数の選択は、次の条件のうち最初に当てはまる条件に基づいて行われます。

  • この関数は、この保存されたモデルのメイン グラフに添付できる唯一の署名です。
  • 関数はデフォルトのシグネチャ名「serving_default」にマップされます。

呼び出し元は、返されたすべての Tensor を閉じる責任があります。

パラメータ
引数シグネチャ名によってマッピングされた入力テンソルのリスト
返品
  • シグネチャ名によってマップされた出力テンソルのリスト
投げる
IllegalArgumentExceptionデフォルトで機能が選択できない場合

public void close ()

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

public static SavedModelBundle.Exporterエクスポーター(String exportDir)

保存したモデルをエクスポートします。

実際にモデルを保存する前に構成オプションを設定するためのExporterオブジェクトを返します。

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

public ConcreteFunction関数(文字列のsignatureKey)

関数シグネチャに対応するConcreteFunctionを返します。

ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
 Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);

パラメータ
署名キー保存されたモデル内のSignatureDefの名前。
返品
  • 関数の呼び出しに使用できるオブジェクト
投げる
IllegalArgumentExceptionこの保存されたモデルにsignatureKeyが見つからない場合。

公開グラフグラフ()

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

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

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

このメソッドは次の短縮形です。

SavedModelBundle.loader().withTags(tags).load();

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

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

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

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

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

public MetaGraphDef metaGraphDef ()

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

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

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

返品
  • 初期化されたセッション

public List< Signature >署名()

この保存されたモデルで使用可能なすべての関数のシグネチャを返します。