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를 구성하는 옵션입니다. |
공개 방법
목록 < 카테고리 > | |
정적 BertNLClassifier | createFromBuffer ( ByteBuffer 모델버퍼) 모델 버퍼와 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 BertNLClassifier 생성합니다. |
정적 BertNLClassifier | createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions 옵션) 모델 버퍼 및 BertNLClassifier.BertNLClassifierOptions 사용하여 BertNLClassifier 생성합니다. |
정적 BertNLClassifier | createFromFile (컨텍스트 컨텍스트, 문자열 modelPath) 메타데이터 및 기본 BertNLClassifier.BertNLClassifierOptions 사용하여 모델 파일에서 BertNLClassifier 생성합니다. |
정적 BertNLClassifier | createFromFile ( 파일 모델파일) |
정적 BertNLClassifier | |
정적 BertNLClassifier | createFromFileAndOptions (컨텍스트 컨텍스트, 문자열 modelPath, BertNLClassifier.BertNLClassifierOptions 옵션) 메타데이터 및 BertNLClassifier.BertNLClassifierOptions 포함된 모델 파일에서 BertNLClassifier 생성합니다. |
상속된 메서드
공개 방법
공개 정적 BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)
모델 버퍼와 기본 BertNLClassifier.BertNLClassifierOptions
사용하여 BertNLClassifier
생성합니다.
매개변수
모델버퍼 | 직접 ByteBuffer 또는 모델의 MappedByteBuffer |
---|
보고
-
BertNLClassifier
인스턴스
던지기
IllegalArgumentException | 모델 버퍼가 직접 ByteBuffer 또는 MappedByteBuffer 가 아닌 경우 |
---|---|
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
공개 정적 BertNLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions 옵션)
모델 버퍼 및 BertNLClassifier.BertNLClassifierOptions
사용하여 BertNLClassifier
생성합니다.
매개변수
모델버퍼 | 직접 ByteBuffer 또는 모델의 MappedByteBuffer |
---|---|
옵션 | 분류자를 구성하려면 |
보고
-
BertNLClassifier
인스턴스
던지기
IllegalArgumentException | 모델 버퍼가 직접 ByteBuffer 또는 MappedByteBuffer 가 아닌 경우 |
---|---|
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
public static BertNLClassifier createFromFile (컨텍스트 컨텍스트, 문자열 modelPath)
메타데이터 및 기본 BertNLClassifier.BertNLClassifierOptions
사용하여 모델 파일에서 BertNLClassifier
생성합니다.
매개변수
문맥 | 안드로이드 컨텍스트 |
---|---|
모델경로 | 분류 모델의 경로 |
보고
-
BertNLClassifier
인스턴스
던지기
IO예외 | 모델 파일을 불러오지 못하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
공개 정적 BertNLClassifier createFromFile ( 파일 모델파일)
메타데이터 및 기본 BertNLClassifier.BertNLClassifierOptions
사용하여 File
객체에서 BertNLClassifier
생성합니다.
매개변수
모델파일 | 분류 모델 File 인스턴스 |
---|
보고
-
BertNLClassifier
인스턴스
던지기
IO예외 | 모델 파일을 불러오지 못하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
public static BertNLClassifier createFromFileAndOptions ( 파일 모델파일, BertNLClassifier.BertNLClassifierOptions 옵션)
메타데이터 및 BertNLClassifier.BertNLClassifierOptions
사용하여 File
객체에서 BertNLClassifier
생성합니다.
매개변수
모델파일 | 분류 모델 File 인스턴스 |
---|---|
옵션 | 분류자를 구성하려면 |
보고
-
BertNLClassifier
인스턴스
던지기
IO예외 | 모델 파일을 불러오지 못하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
public static BertNLClassifier createFromFileAndOptions (컨텍스트 컨텍스트, 문자열 modelPath, BertNLClassifier.BertNLClassifierOptions 옵션)
메타데이터 및 BertNLClassifier.BertNLClassifierOptions
사용하여 모델 파일에서 BertNLClassifier
생성합니다.
매개변수
문맥 | 안드로이드 컨텍스트. |
---|---|
모델경로 | 분류 모델의 경로 |
옵션 | 분류자를 구성하려면 |
보고
-
BertNLClassifier
인스턴스
던지기
IO예외 | 모델 파일을 불러오지 못하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |