SavedModelBundle reprezentuje model załadowany z magazynu.
Model składa się z opisu obliczeń ( Graph ), Session z tensorami (np. parametrami lub zmiennymi na wykresie) zainicjowanymi do wartości zapisanych w pamięci oraz opisu modelu jako bufora protokołu MetaGraphDef .
Klasy zagnieżdżone
| klasa | SavedModelBundle.Exporter | Opcje eksportowania SavedModel. | |
| klasa | SavedModelBundle.Loader | Opcje ładowania SavedModel. | |
Stałe
| Smyczkowy | DOMYŚLNY_TAG |
Metody publiczne
| Mapa<String, Tensor > | |
| próżnia | |
| statyczny SavedModelBundle.Exporter | eksporter (String eksportDir) Eksportuj zapisany model. |
| Funkcja betonu | funkcja (String podpisKey) Zwróć ConcreteFunction odpowiadającą sygnaturze funkcji. |
| Wykres | wykres () Zwraca wykres opisujący obliczenia wykonane przez model. |
| statyczny plik SavedModelBundle | załaduj (String eksportDir, String... tagi) Załaduj zapisany model z katalogu eksportu. |
| statyczny SavedModelBundle.Loader | moduł ładujący (String eksportDir) Załaduj zapisany model. |
| MetaGraphDef | metaGraphDef () Zwraca bufor protokołu MetaGraphDef powiązany z zapisanym modelem. |
| Sesja | |
| Lista <Podpis> | podpisy () Zwróć sygnaturę wszystkich funkcji dostępnych w tym zapisanym modelu. |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków DEFAULT_TAG
Metody publiczne
public Map<String, Tensor > wywołanie (Map<String, Tensor > argumenty)
Wywołuje funkcję domyślną bezpośrednio z tego modelu.
Domyślny wybór funkcji odbywa się na podstawie pierwszego z poniższych warunków, który jest spełniony:
- Funkcja jest jedyną dostępną sygnaturą dołączoną do głównego wykresu zapisanego modelu
- Funkcja jest mapowana na domyślną nazwę podpisu, czyli „serving_default”
Osoba dzwoniąca jest odpowiedzialna za zamknięcie wszystkich zwróconych tensorów.
Parametry
| argumenty | lista tensorów wejściowych, odwzorowana według ich nazwy podpisu |
|---|
Powroty
- lista tensorów wyjściowych, odwzorowanych według nazwy podpisu
Rzuca
| Wyjątek IllegalArgument | jeśli domyślnie nie można wybrać żadnej funkcji |
|---|
publiczna pustka zamknij ()
public static SavedModelBundle.Exporter eksporter (String eksportDir)
Eksportuj zapisany model.
Zwraca obiekt Exporter do ustawiania opcji konfiguracyjnych przed faktycznym zapisaniem modelu.
Parametry
| eksportDir | ścieżka katalogu zawierającego zapisany model. |
|---|
publiczna funkcja ConcreteFunction (String podpisKey)
Zwróć ConcreteFunction odpowiadającą sygnaturze funkcji.
ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
Parametry
| podpisKlucz | nazwa SignatureDef w zapisanym modelu. |
|---|
Powroty
- obiekt, którego można użyć do wywołania funkcji
Rzuca
| Wyjątek IllegalArgument | jeśli w zapisanym modelu nie zostanie znaleziony signatureKey . |
|---|
publiczne statyczne ładowanie pakietu SavedModelBundle (tagi String eksportDir, String...)
Załaduj zapisany model z katalogu eksportu. Ładowany model należy utworzyć przy użyciu API Saved Model .
Ta metoda jest skrótem od:
SavedModelBundle.loader().withTags(tags).load();
Parametry
| eksportDir | ścieżka katalogu zawierającego zapisany model. |
|---|---|
| tagi | znaczniki identyfikujące konkretny metagraphdef do załadowania. |
Powroty
- pakiet zawierający wykres i powiązaną sesję.
public static SavedModelBundle.Loader moduł ładujący (String eksportDir)
Załaduj zapisany model.
Zwraca obiekt Loader , który może ustawić opcje konfiguracyjne przed faktycznym załadowaniem modelu,
Parametry
| eksportDir | ścieżka katalogu zawierającego zapisany model. |
|---|
publiczny MetaGraphDef metaGraphDef ()
Zwraca bufor protokołu MetaGraphDef powiązany z zapisanym modelem.
sesja publiczna sesja ()
Zwraca Session , za pomocą której należy wykonać obliczenia przy użyciu modelu.
Powroty
- zainicjowanej sesji
publiczna Lista< Podpis > podpisy ()
Zwróć sygnaturę wszystkich funkcji dostępnych w tym zapisanym modelu.