BertNLClassifier

공개 클래스 BertNLClassifier

Bert 모델을 사용한 NLClassification 작업용 분류자 API는 문자열을 다양한 클래스로 분류합니다. API는 메타데이터가 채워진 Bert 기반 TFLite 모델을 기대합니다.

메타데이터에는 다음 정보가 포함되어야 합니다.

  • Wordpiece/Sentencepiece Tokenizer용 input_process_unit 1개.
  • 이름이 "ids", "mask" 및 "segment_ids"인 입력 텐서 3개.
  • 선택적으로 라벨 파일이 첨부된 float32[1, 2] 유형의 출력 텐서 1개. 레이블 파일이 첨부된 경우 파일은 한 줄에 하나의 레이블이 있는 일반 텍스트 파일이어야 하며 레이블 수는 모델이 출력하는 범주 수와 일치해야 합니다.

중첩 클래스

수업 BertNLClassifier.BertNLClassifierOptions BertNLClassifier를 구성하는 옵션입니다.

공개 방법

목록 < 카테고리 >
분류 ( 문자열 텍스트)
문자열 입력에 대해 분류를 수행하고 분류된 Category 를 반환합니다.
정적 BertNLClassifier
createFromBuffer ( ByteBuffer 모델버퍼)
모델 버퍼와 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 BertNLClassifier 생성합니다.
정적 BertNLClassifier
createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions 옵션)
모델 버퍼 및 BertNLClassifier.BertNLClassifierOptions 사용하여 BertNLClassifier 생성합니다.
정적 BertNLClassifier
createFromFile (컨텍스트 컨텍스트, 문자열 modelPath)
메타데이터 및 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 모델 파일에서 BertNLClassifier 생성합니다.
정적 BertNLClassifier
createFromFile ( 파일 모델파일)
메타데이터 및 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 File 객체에서 BertNLClassifier 생성합니다.
정적 BertNLClassifier
createFromFileAndOptions ( 파일 모델파일, BertNLClassifier.BertNLClassifierOptions 옵션)
메타데이터 및 BertNLClassifier.BertNLClassifierOptions 사용하여 File 객체에서 BertNLClassifier 생성합니다.
정적 BertNLClassifier
createFromFileAndOptions (컨텍스트 컨텍스트, 문자열 modelPath, BertNLClassifier.BertNLClassifierOptions 옵션)
메타데이터 및 BertNLClassifier.BertNLClassifierOptions 포함된 모델 파일에서 BertNLClassifier 생성합니다.

상속된 메서드

공개 방법

공개 목록 < 카테고리 > 분류 ( 문자열 텍스트)

문자열 입력에 대해 분류를 수행하고 분류된 Category 를 반환합니다.

매개변수
텍스트 모델에 텍스트를 입력합니다.
보고
  • 카테고리 결과 목록입니다.

공개 정적 BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

모델 버퍼와 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 BertNLClassifier 생성합니다.

매개변수
모델버퍼 직접 ByteBuffer 또는 모델의 MappedByteBuffer
보고
던지기
IllegalArgumentException 모델 버퍼가 직접 ByteBuffer 또는 MappedByteBuffer 가 아닌 경우
IllegalStateException 내부 오류가 있는 경우
런타임예외 달리 명시되지 않은 오류가 있는 경우

공개 정적 BertNLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions 옵션)

모델 버퍼 및 BertNLClassifier.BertNLClassifierOptions 사용하여 BertNLClassifier 생성합니다.

매개변수
모델버퍼 직접 ByteBuffer 또는 모델의 MappedByteBuffer
옵션 분류자를 구성하려면
보고
던지기
IllegalArgumentException 모델 버퍼가 직접 ByteBuffer 또는 MappedByteBuffer 가 아닌 경우
IllegalStateException 내부 오류가 있는 경우
런타임예외 달리 명시되지 않은 오류가 있는 경우

public static BertNLClassifier createFromFile (컨텍스트 컨텍스트, 문자열 modelPath)

메타데이터 및 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 모델 파일에서 BertNLClassifier 생성합니다.

매개변수
문맥 안드로이드 컨텍스트
모델경로 분류 모델의 경로
보고
던지기
IO예외 모델 파일을 불러오지 못하는 경우
IllegalArgumentException 인수가 유효하지 않은 경우
IllegalStateException 내부 오류가 있는 경우
런타임예외 달리 명시되지 않은 오류가 있는 경우

공개 정적 BertNLClassifier createFromFile ( 파일 모델파일)

메타데이터 및 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 File 객체에서 BertNLClassifier 생성합니다.

매개변수
모델파일 분류 모델 File 인스턴스
보고
던지기
IO예외 모델 파일을 불러오지 못하는 경우
IllegalArgumentException 인수가 유효하지 않은 경우
IllegalStateException 내부 오류가 있는 경우
런타임예외 달리 명시되지 않은 오류가 있는 경우

public static BertNLClassifier createFromFileAndOptions ( 파일 모델파일, BertNLClassifier.BertNLClassifierOptions 옵션)

메타데이터 및 BertNLClassifier.BertNLClassifierOptions 사용하여 File 객체에서 BertNLClassifier 생성합니다.

매개변수
모델파일 분류 모델 File 인스턴스
옵션 분류자를 구성하려면
보고
던지기
IO예외 모델 파일을 불러오지 못하는 경우
IllegalArgumentException 인수가 유효하지 않은 경우
IllegalStateException 내부 오류가 있는 경우
런타임예외 달리 명시되지 않은 오류가 있는 경우

public static BertNLClassifier createFromFileAndOptions (컨텍스트 컨텍스트, 문자열 modelPath, BertNLClassifier.BertNLClassifierOptions 옵션)

메타데이터 및 BertNLClassifier.BertNLClassifierOptions 사용하여 모델 파일에서 BertNLClassifier 생성합니다.

매개변수
문맥 안드로이드 컨텍스트.
모델경로 분류 모델의 경로
옵션 분류자를 구성하려면
보고
던지기
IO예외 모델 파일을 불러오지 못하는 경우
IllegalArgumentException 인수가 유효하지 않은 경우
IllegalStateException 내부 오류가 있는 경우
런타임예외 달리 명시되지 않은 오류가 있는 경우