パブリック クラスSavedModelBundle
SavedModelBundle は、ストレージからロードされたモデルを表します。
モデルは、計算の記述 ( Graph )、ストレージに保存された値に初期化されたSession (グラフ内のパラメーターや変数など) を含むセッション、およびMetaGraphDef プロトコル バッファーとしてのモデルの記述で構成されます。
ネストされたクラス
| クラス | SavedModelBundle.Exporter | SavedModel をエクスポートするためのオプション。 | |
| クラス | SavedModelBundle.Loader | SavedModel をロードするためのオプション。 | |
定数
| 弦 | DEFAULT_TAG |
パブリックメソッド
| Map<String, Tensor > | |
| 空所 | |
| 静的SavedModelBundle.Exporter | エクスポータ(文字列のexportDir) 保存したモデルをエクスポートします。 |
| 具体的な関数 | 関数(文字列の署名キー) 関数シグネチャに対応する ConcreteFunctionを返します。 |
| グラフ | グラフ（） モデルによって実行された計算を説明するグラフを返します。 |
| 静的SavedModelBundle | ロード(文字列エクスポートディレクトリ、文字列... タグ) 保存されたモデルをエクスポート ディレクトリからロードします。 |
| 静的SavedModelBundle.Loader | ローダー(文字列のエクスポートディレクトリ) 保存したモデルをロードします。 |
| メタグラフ定義 | メタグラフ定義() 保存されたモデルに関連付けられたMetaGraphDef プロトコル バッファを返します。 |
| セッション | |
| リスト<署名> | 署名() この保存されたモデルで使用可能なすべての関数のシグネチャを返します。 |
継承されたメソッド
定数
パブリック静的最終文字列DEFAULT_TAG
定数値: 「サーブ」
パブリックメソッド
public Map<String, Tensor > call (Map<String, Tensor > 引数)
このモデルからデフォルト関数を直接呼び出します。
デフォルトの関数の選択は、次の条件のうち最初に当てはまる条件に基づいて行われます。
- この関数は、この保存されたモデルのメイン グラフに添付できる唯一の署名です。
- 関数はデフォルトのシグネチャ名「serving_default」にマップされます。
呼び出し元は、返されたすべての Tensor を閉じる責任があります。
パラメータ
| 引数 | シグネチャ名によってマッピングされた入力テンソルのリスト |
|---|
返品
- シグネチャ名によってマッピングされた出力テンソルのリスト
投げる
| IllegalArgumentException | デフォルトで機能が選択できない場合 |
|---|
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オブジェクトを返します。
パラメータ
| エクスポートディレクトリ | 保存されたモデルを含むディレクトリ パス。 |
|---|