แยกเอนทิตีด้วย ML Kit บน Android

หากต้องการวิเคราะห์ข้อความและดึงข้อมูลเอนทิตีในข้อความ ให้เรียกใช้เมธอด annotate() แล้วส่งสตริงข้อความหรืออินสแตนซ์ของ EntityExtractionParams ซึ่งระบุตัวเลือกเพิ่มเติมได้ เช่น เวลาอ้างอิง เขตเวลา หรือตัวกรองเพื่อจำกัดการค้นหาเอนทิตีประเภทย่อย API จะแสดงรายการออบเจ็กต์ EntityAnnotation ที่มีข้อมูลเกี่ยวกับเอนทิตีแต่ละรายการ

ชื่อ SDKentity-extraction
การใช้งานชิ้นงานสำหรับเครื่องตรวจจับพื้นฐานจะลิงก์กับแอปแบบคงที่ ณ เวลาที่สร้าง
ผลกระทบจากขนาดชิ้นงานการสกัดเอนทิตีจะส่งผลต่อขนาดแอปได้สูงสุดประมาณ 5.6 MB

ลองเลย

ก่อนเริ่มต้น

  1. ในไฟล์ build.gradle ระดับโปรเจ็กต์ ให้ตรวจสอบว่าที่เก็บ Maven ของ Google รวมอยู่ทั้งในส่วน buildscript และ allprojects

  2. เพิ่มทรัพยากร Dependency สำหรับไลบรารีการดึงข้อมูลเอนทิตีของ ML Kit ลงในไฟล์ Gradle ระดับแอปของโมดูล ซึ่งมักจะมีชื่อว่า app/build.gradle โดยทำดังนี้

    dependencies {
// …

implementation 'com.google.mlkit:entity-extraction:16.0.0-beta5'
}

แยกเอนทิตี

สร้างออบเจ็กต์ EntityExtractor และกำหนดค่าด้วย EntityExtractorOptions

Kotlin

val entityExtractor =
   EntityExtraction.getClient(
       EntityExtractorOptions.Builder(EntityExtractorOptions.ENGLISH)
           .build())

Java

EntityExtractor entityExtractor =
        EntityExtraction.getClient(
            new EntityExtractorOptions.Builder(EntityExtractorOptions.ENGLISH)
                .build());

EntityExtractorOptions ยังยอมรับ Executor ที่ผู้ใช้กําหนดด้วยหากต้องการ ไม่เช่นนั้นจะใช้ Executor เริ่มต้นใน ML Kit

ตรวจสอบว่าได้ดาวน์โหลดโมเดลที่จำเป็นลงในอุปกรณ์แล้ว

Kotlin

entityExtractor
  .downloadModelIfNeeded()
  .addOnSuccessListener { _ ->
    /* Model downloading succeeded, you can call extraction API here. */
  }
  .addOnFailureListener { _ -> /* Model downloading failed. */ }

Java

entityExtractor
    .downloadModelIfNeeded()
    .addOnSuccessListener(
        aVoid -> {
          // Model downloading succeeded, you can call the extraction API here. 
        })
    .addOnFailureListener(
        exception -> {
          // Model downloading failed.
        });

หลังจากยืนยันว่าดาวน์โหลดโมเดลแล้ว ให้ส่งสตริงหรือ EntityExtractionParams ไปยัง annotate() อย่าโทรหา annotate() จนกว่าคุณจะทราบว่ารุ่นดังกล่าวพร้อมจำหน่ายแล้ว

Kotlin

val params =
      EntityExtractionParams.Builder("My flight is LX373, please pick me up at 8am tomorrow.")
        .setEntityTypesFilter((/* optional entity type filter */)
        .setPreferredLocale(/* optional preferred locale */)
        .setReferenceTime(/* optional reference date-time */)
        .setReferenceTimeZone(/* optional reference timezone */)
        .build()
entityExtractor
      .annotate(params)
      .addOnSuccessListener {
        // Annotation process was successful, you can parse the EntityAnnotations list here.
      }
      .addOnFailureListener {
        // Check failure message here.
      }

Java

EntityExtractionParams params = new EntityExtractionParams
        .Builder("My flight is LX373, please pick me up at 8am tomorrow.")
        .setEntityTypesFilter(/* optional entity type filter */)
        .setPreferredLocale(/* optional preferred locale */)
        .setReferenceTime(/* optional reference date-time */)
        .setReferenceTimeZone(/* optional reference timezone */)
        .build();
entityExtractor
        .annotate(params)
        .addOnSuccessListener(new OnSuccessListener<List<EntityAnnotation>>() {
          @Override
          public void onSuccess(List<EntityAnnotation> entityAnnotations) {
            // Annotation process was successful, you can parse the EntityAnnotations list here.
          }
        })
        .addOnFailureListener(new OnFailureListener() {
          @Override
          public void onFailure(@NonNull Exception e) {
            // Check failure message here.
          }
        });

PreferredLocale, ReferenceTime และ ReferenceTimeZone จะส่งผลต่อเอนทิตี DateTime เท่านั้น หากไม่ได้ตั้งค่าเหล่านี้อย่างชัดเจน ระบบจะใช้ค่าเริ่มต้นจากอุปกรณ์ของผู้ใช้

วนดูผลลัพธ์คำอธิบายประกอบเพื่อเรียกข้อมูลเกี่ยวกับเอนทิตีที่รู้จัก

Kotlin

for (entityAnnotation in entityAnnotations) {
  val entities: List<Entity> = entityAnnotation.entities

  Log.d(TAG, "Range: ${entityAnnotation.start} - ${entityAnnotation.end}")
  for (entity in entities) {
    when (entity) {
      is DateTimeEntity -> {
        Log.d(TAG, "Granularity: ${entity.dateTimeGranularity}")
        Log.d(TAG, "TimeStamp: ${entity.timestampMillis}")
      }
      is FlightNumberEntity -> {
        Log.d(TAG, "Airline Code: ${entity.airlineCode}")
        Log.d(TAG, "Flight number: ${entity.flightNumber}")
      }
      is MoneyEntity -> {
        Log.d(TAG, "Currency: ${entity.unnormalizedCurrency}")
        Log.d(TAG, "Integer part: ${entity.integerPart}")
        Log.d(TAG, "Fractional Part: ${entity.fractionalPart}")
      }
      else -> {
        Log.d(TAG, "  $entity")
      }
    }
  }
}

Java

List<EntityAnnotation> entityAnnotations = /* Get from EntityExtractor */;
for (EntityAnnotation entityAnnotation : entityAnnotations) {
  List<Entity> entities = entityAnnotation.getEntities();

  Log.d(TAG, String.format("Range: [%d, %d)", entityAnnotation.getStart(), entityAnnotation.getEnd()));
  for (Entity entity : entities) {
    switch (entity.getType()) {
      case Entity.TYPE_DATE_TIME:
        DateTimeEntity dateTimeEntity = entity.asDateTimeEntity();
        Log.d(TAG, "Granularity: " + dateTimeEntity.getDateTimeGranularity());
        Log.d(TAG, "Timestamp: " + dateTimeEntity.getTimestampMillis());
      case Entity.TYPE_FLIGHT_NUMBER:
        FlightNumberEntity flightNumberEntity = entity.asFlightNumberEntity();
        Log.d(TAG, "Airline Code: " + flightNumberEntity.getAirlineCode());
        Log.d(TAG, "Flight number: " + flightNumberEntity.getFlightNumber());
      case Entity.TYPE_MONEY:
        MoneyEntity moneyEntity = entity.asMoneyEntity();
        Log.d(TAG, "Currency: " + moneyEntity.getUnnormalizedCurrency());
        Log.d(TAG, "Integer Part: " + moneyEntity.getIntegerPart());
        Log.d(TAG, "Fractional Part: " + moneyEntity.getFractionalPart());
      case Entity.TYPE_UNKNOWN:
      default:
        Log.d(TAG, "Entity: " + entity);
    }
  }
}

เรียกใช้เมธอด close() เมื่อไม่ต้องการออบเจ็กต์ EntityExtractor อีกต่อไป หากใช้ EntityExtractor ใน Fragment หรือ AppCompatActivity คุณสามารถเรียก LifecycleOwner.getLifecycle() ใน Fragment หรือ AppCompatActivity แล้วเรียก Lifecycle.addObserver เช่น

Kotlin

val options = …
val extractor = EntityExtraction.getClient(options);
getLifecycle().addObserver(extractor);

Java

EntityExtractorOptions options = …
EntityExtractor extractor = EntityExtraction.getClient(options);
getLifecycle().addObserver(extractor);

จัดการโมเดลการดึงข้อมูลเอนทิตีอย่างชัดเจน

เมื่อคุณใช้ API การดึงข้อมูลเอนทิตีตามที่อธิบายไว้ข้างต้น ML Kit จะดาวน์โหลดโมเดลเฉพาะภาษาลงในอุปกรณ์โดยอัตโนมัติตามที่จำเป็น (เมื่อคุณเรียกใช้ downloadModelIfNeeded()) นอกจากนี้ คุณยังจัดการโมเดลที่ต้องการให้พร้อมใช้งานบนอุปกรณ์ได้อย่างชัดเจนโดยใช้ API การจัดการโมเดลของ ML Kit ซึ่งจะเป็นประโยชน์หากคุณต้องการดาวน์โหลดโมเดลล่วงหน้า นอกจากนี้ API ยังช่วยให้คุณลบโมเดลที่ไม่จําเป็นต้องใช้แล้วออกได้

Kotlin

val modelManager = RemoteModelManager.getInstance()

// Get entity extraction models stored on the device.
modelManager.getDownloadedModels(EntityExtractionRemoteModel::class.java)
  .addOnSuccessListener {
    // ...
  }
  .addOnFailureListener({
    // Error.
  })
    
// Delete the German model if it's on the device.
val germanModel =
  EntityExtractionRemoteModel.Builder(EntityExtractorOptions.GERMAN).build()
modelManager.deleteDownloadedModel(germanModel)
  .addOnSuccessListener({
    // Model deleted.
  })
  .addOnFailureListener({
    // Error.
  })
    
// Download the French model.
val frenchModel =
  EntityExtractionRemoteModel.Builder(EntityExtractorOptions.FRENCH).build()
val conditions =
  DownloadConditions.Builder()
    .requireWifi()
    .build()
modelManager.download(frenchModel, conditions)
  .addOnSuccessListener({
    // Model downloaded.
  })
  .addOnFailureListener({
    // Error.
  })

Java

// Get entity extraction models stored on the device.
modelManager.getDownloadedModels(EntityExtractionRemoteModel.class)
    .addOnSuccessListener(new OnSuccessListener<Set<EntityExtractionRemoteModel>>() {
      @Override
      public void onSuccess(Set<EntityExtractionRemoteModel> models) {
        // ...
      }
    })
    .addOnFailureListener(new OnFailureListener() {
      @Override
      public void onFailure(@NonNull Exception e) {
        // Error.
      }
    });

// Delete the German model if it's on the device.
EntityExtractionRemoteModel germanModel = new EntityExtractionRemoteModel.Builder(EntityExtractorOptions.GERMAN).build();
    modelManager.deleteDownloadedModel(germanModel)
        .addOnSuccessListener(new OnSuccessListener<Void>() {
          @Override
          public void onSuccess(Void v) {
            // Model deleted.
          }
        })
        .addOnFailureListener(new OnFailureListener() {
          @Override
          public void onFailure(@NonNull Exception e) {
            // Error.
          }
        });

// Download the French model.
EntityExtractionRemoteModel frenchModel = new EntityExtractionRemoteModel.Builder(EntityExtractorOptions.FRENCH).build();
    DownloadConditions conditions = new DownloadConditions.Builder()
        .requireWifi()
        .build();
    modelManager.download(frenchModel, conditions)
        .addOnSuccessListener(new OnSuccessListener<Void>() {
          @Override
          public void onSuccess(Void v) {
            // Model downloaded.
          }
        })
        .addOnFailureListener(new OnFailureListener() {
          @Override
          public void onFailure(@NonNull Exception e) {
            // Error.
          }
        });