SavedModelBundle

공개 클래스 SavedModelBundle

SavedModelBundle은 저장소에서 로드된 모델을 나타냅니다.

모델은 계산에 대한 설명( Graph ), 스토리지에 저장된 값으로 초기화된 텐서(예: 그래프의 매개변수 또는 변수)가 있는 Session , MetaGraphDef 프로토콜 버퍼 로서의 모델에 대한 설명으로 구성됩니다.

중첩 클래스

수업 저장된ModelBundle.Exporter 저장된 모델을 내보내기 위한 옵션입니다.
수업 저장된ModelBundle.Loader 저장된 모델을 로드하기 위한 옵션입니다.

상수

DEFAULT_TAG

공개 방법

맵<문자열, 텐서 >
호출 (Map<String, Tensor > 인수)
이 모델에서 직접 기본 기능을 호출합니다.
무효의
닫다 ()
저장된 모델 번들과 관련된 리소스( GraphSession )를 해제합니다.
정적 SavedModelBundle.Exporter
내보내기 (문자열 내보내기Dir)
저장된 모델을 내보냅니다.
콘크리트함수
함수 (문자열 서명키)
함수 시그니처에 해당하는 ConcreteFunction 을 반환합니다.
그래프
그래프 ()
모델이 수행한 계산을 설명하는 그래프를 반환합니다.
정적 저장된 모델 번들
로드 (문자열 내보내기Dir, 문자열... 태그)
내보내기 디렉터리에서 저장된 모델을 로드합니다.
정적 SavedModelBundle.Loader
로더 (문자열 내보내기Dir)
저장된 모델을 로드합니다.
메타그래프Def
메타그래프Def ()
저장된 모델과 연관된 MetaGraphDef 프로토콜 버퍼를 반환합니다.
세션
세션 ()
모델을 사용하여 계산을 수행할 Session 반환합니다.
목록< 서명 >
서명 ()
이 저장된 모델에서 사용할 수 있는 모든 함수의 서명을 반환합니다.

상속된 메서드

상수

공개 정적 최종 문자열 DEFAULT_TAG

상수 값: "서브"

공개 방법

공개 Map<String, Tensor > 호출 (Map<String, Tensor > 인수)

이 모델에서 직접 기본 기능을 호출합니다.

기본 기능 선택은 다음 조건 중 첫 번째가 참인 경우에 따라 수행됩니다.

  • 이 기능은 저장된 모델의 기본 그래프에 첨부할 수 있는 유일한 서명입니다.
  • 함수는 기본 서명 이름인 "serving_default"에 매핑됩니다.

호출자는 반환된 모든 Tensor를 닫을 책임이 있습니다.

매개변수
인수 서명 이름으로 매핑된 입력 텐서 목록
보고
  • 서명 이름으로 매핑된 출력 텐서 목록
던지기
IllegalArgumentException 기본적으로 기능을 선택할 수 없는 경우

공공 무효 닫기 ()

저장된 모델 번들과 관련된 리소스( GraphSession )를 해제합니다.

공개 정적 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 객체를 반환합니다.

매개변수
내보내기 디렉터리 저장된 모델이 포함된 디렉터리 경로입니다.

공개 MetaGraphDef MetaGraphDef ()

저장된 모델과 연관된 MetaGraphDef 프로토콜 버퍼를 반환합니다.

공개 세션 세션 ()

모델을 사용하여 계산을 수행할 Session 반환합니다.

보고
  • 초기화된 세션

공개 목록< 서명 > 서명 ()

저장된 모델에서 사용할 수 있는 모든 함수의 서명을 반환합니다.