공개 클래스 SavedModelBundle
SavedModelBundle은 저장소에서 로드된 모델을 나타냅니다.
모델은 계산에 대한 설명( Graph ), 스토리지에 저장된 값으로 초기화된 텐서(예: 그래프의 매개변수 또는 변수)가 있는 Session , MetaGraphDef 프로토콜 버퍼 로서의 모델에 대한 설명으로 구성됩니다.
중첩 클래스
| 수업 | 저장된ModelBundle.Exporter | 저장된 모델을 내보내기 위한 옵션입니다. | |
| 수업 | 저장된ModelBundle.Loader | 저장된 모델을 로드하기 위한 옵션입니다. | |
상수
| 끈 | DEFAULT_TAG |
공개 방법
| 맵<문자열, 텐서 > | |
| 무효의 | |
| 정적 SavedModelBundle.Exporter | 내보내기 (문자열 내보내기Dir) 저장된 모델을 내보냅니다. |
| 콘크리트함수 | 함수 (문자열 서명키) 함수 시그니처에 해당하는 ConcreteFunction 반환합니다. |
| 그래프 | 그래프 () 모델이 수행한 계산을 설명하는 그래프를 반환합니다. |
| 정적 저장된 모델 번들 | 로드 (문자열 내보내기Dir, 문자열... 태그) 내보내기 디렉터리에서 저장된 모델을 로드합니다. |
| 정적 SavedModelBundle.Loader | 로더 (문자열 내보내기Dir) 저장된 모델을 로드합니다. |
| 메타그래프Def | 메타그래프Def () 저장된 모델과 연관된 MetaGraphDef 프로토콜 버퍼를 반환합니다. |
| 세션 | |
| 목록< 서명 > | 서명 () 이 저장된 모델에서 사용할 수 있는 모든 함수의 서명을 반환합니다. |
상속된 메서드
상수
공개 정적 최종 문자열 DEFAULT_TAG
상수 값: "서브"
공개 방법
공개 Map<String, Tensor > 호출 (Map<String, Tensor > 인수)
이 모델에서 직접 기본 기능을 호출합니다.
기본 기능 선택은 다음 조건 중 첫 번째가 참인 경우에 따라 수행됩니다.
- 이 기능은 저장된 모델의 기본 그래프에 첨부할 수 있는 유일한 서명입니다.
- 함수는 기본 서명 이름인 "serving_default"에 매핑됩니다.
호출자는 반환된 모든 Tensor를 닫을 책임이 있습니다.
매개변수
| 인수 | 서명 이름으로 매핑된 입력 텐서 목록 |
|---|
보고
- 서명 이름으로 매핑된 출력 텐서 목록
던지기
| IllegalArgumentException | 기본적으로 기능을 선택할 수 없는 경우 |
|---|
공개 정적 SavedModelBundle.Exporter 내보내기 (문자열 내보내기Dir)
저장된 모델을 내보냅니다.
실제로 모델을 저장하기 전에 구성 옵션을 설정하기 위한 Exporter 개체를 반환합니다.
매개변수
| 내보내기 디렉터리 | 저장된 모델이 포함된 디렉터리 경로입니다. |
|---|
공용 ConcreteFunction 함수 (문자열 서명 키)
함수 시그니처에 해당하는 ConcreteFunction 반환합니다.
ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
매개변수
| 서명키 | 저장된 모델의 SignatureDef 이름입니다. |
|---|
보고
- 함수를 호출하는 데 사용할 수 있는 객체
던지기
| IllegalArgumentException | 저장된 모델에서 signatureKey 찾을 수 없는 경우. |
|---|
공개 정적 SavedModelBundle 로드 (문자열 내보내기Dir, 문자열... 태그)
내보내기 디렉터리에서 저장된 모델을 로드합니다. 로드되는 모델은 저장된 모델 API를 사용하여 생성되어야 합니다.
이 방법은 다음을 간략하게 표현한 것입니다.
SavedModelBundle.loader().withTags(tags).load();
매개변수
| 내보내기 디렉터리 | 저장된 모델이 포함된 디렉터리 경로입니다. |
|---|---|
| 태그 | 로드할 특정 Metagraphdef를 식별하는 태그입니다. |
보고
- 그래프와 관련 세션이 포함된 번들입니다.
공개 정적 SavedModelBundle.Loader 로더 (문자열 내보내기Dir)
저장된 모델을 로드합니다.
실제로 모델을 로드하기 전에 구성 옵션을 설정할 수 있는 Loader 객체를 반환합니다.
매개변수
| 내보내기 디렉터리 | 저장된 모델이 포함된 디렉터리 경로입니다. |
|---|