SavedModelBundle

SavedModelBundle kelas publik

SavedModelBundle mewakili model yang dimuat dari penyimpanan.

Model terdiri dari deskripsi komputasi ( Graph ), Session dengan tensor (misalnya parameter atau variabel dalam grafik) yang diinisialisasi ke nilai yang disimpan dalam penyimpanan, dan deskripsi model sebagai buffer protokol MetaGraphDef .

Kelas Bersarang

kelas SavedModelBundle.Eksportir Opsi untuk mengekspor SavedModel.
kelas SavedModelBundle.Loader Opsi untuk memuat SavedModel.

Konstanta

Rangkaian DEFAULT_TAG

Metode Publik

Peta<String, Tensor >
panggilan (argumen Map<String, Tensor >)
Memanggil fungsi default langsung dari model ini.
ruang kosong
menutup ()
Merilis sumber daya ( Graph dan Session ) yang terkait dengan paket model tersimpan.
SavedModelBundle.Exporter statis
eksportir (String eksporDir)
Ekspor model yang disimpan.
Fungsi Beton
fungsi (String kunci tanda tangan)
Mengembalikan ConcreteFunction yang sesuai dengan tanda tangan fungsi.
Grafik
grafik ()
Mengembalikan grafik yang menjelaskan komputasi yang dilakukan oleh model.
SavedModelBundle statis
memuat (String eksporDir, String... tag)
Muat model yang disimpan dari direktori ekspor.
SavedModelBundle.Loader statis
pemuat (String eksporDir)
Muat model yang disimpan.
MetaGraphDef
metaGraphDef ()
Mengembalikan buffer protokol MetaGraphDef yang terkait dengan model yang disimpan.
Sidang
sesi ()
Mengembalikan Session untuk melakukan komputasi menggunakan model.
Daftar< Tanda tangan >
tanda tangan ()
Kembalikan tanda tangan semua fungsi yang tersedia dalam model tersimpan ini.

Metode Warisan

Konstanta

String akhir statis publik DEFAULT_TAG

Nilai Konstan: "melayani"

Metode Publik

panggilan publik Map<String, Tensor > (argumen Map<String, Tensor >)

Memanggil fungsi default langsung dari model ini.

Pemilihan fungsi default dilakukan berdasarkan kondisi pertama berikut ini yang benar:

  • Fungsi tersebut adalah satu-satunya tanda tangan yang tersedia yang dilampirkan pada grafik utama model tersimpan ini
  • Fungsi tersebut dipetakan ke nama tanda tangan default, yaitu "serving_default"

Penelepon bertanggung jawab untuk menutup semua Tensor yang dikembalikan.

Parameter
argumen daftar tensor masukan, dipetakan berdasarkan nama tanda tangannya
Kembali
  • daftar tensor keluaran, dipetakan berdasarkan nama tanda tangan
Melempar
Pengecualian Argumen Ilegal jika tidak ada fungsi yang dapat dipilih secara default

penutupan kekosongan publik ()

Merilis sumber daya ( Graph dan Session ) yang terkait dengan paket model tersimpan.

eksportir SavedModelBundle.Exporter statis publik (String eksporDir)

Ekspor model yang disimpan.

Mengembalikan objek Exporter untuk menyetel opsi konfigurasi sebelum benar-benar menyimpan model.

Parameter
eksporDir jalur direktori yang berisi model yang disimpan.

fungsi ConcreteFunction publik (String SignatureKey)

Mengembalikan ConcreteFunction yang sesuai dengan tanda tangan fungsi.

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

Parameter
tanda tanganKey nama SignatureDef dalam model yang disimpan.
Kembali
  • objek yang dapat digunakan untuk melakukan panggilan ke suatu fungsi
Melempar
Pengecualian Argumen Ilegal jika signatureKey tidak ditemukan dalam model tersimpan ini.

grafik Grafik publik ()

Mengembalikan grafik yang menjelaskan komputasi yang dilakukan oleh model.

pemuatan SavedModelBundle statis publik (String eksporDir, String... tag)

Muat model yang disimpan dari direktori ekspor. Model yang sedang dimuat harus dibuat menggunakan Saved Model API .

Metode ini merupakan singkatan dari:

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

Parameter
eksporDir jalur direktori yang berisi model yang disimpan.
tag tag yang mengidentifikasi metagraphdef spesifik yang akan dimuat.
Kembali
  • bundel yang berisi grafik dan sesi terkait.

Pemuat SavedModelBundle.Loader statis publik (String eksporDir)

Muat model yang disimpan.

Mengembalikan objek Loader yang dapat mengatur opsi konfigurasi sebelum benar-benar memuat model,

Parameter
eksporDir jalur direktori yang berisi model yang disimpan.

MetaGraphDef publik metaGraphDef ()

Mengembalikan buffer protokol MetaGraphDef yang terkait dengan model yang disimpan.

sesi Sesi publik ()

Mengembalikan Session untuk melakukan komputasi menggunakan model.

Kembali
  • sesi yang diinisialisasi

daftar publik< Tanda tangan > tanda tangan ()

Kembalikan tanda tangan semua fungsi yang tersedia dalam model tersimpan ini.