/
ComposeDocumentModelAsync.java
89 lines (76 loc) · 4.58 KB
/
ComposeDocumentModelAsync.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.ai.formrecognizer.administration;
import com.azure.ai.formrecognizer.documentanalysis.administration.DocumentModelAdministrationAsyncClient;
import com.azure.ai.formrecognizer.documentanalysis.administration.DocumentModelAdministrationClientBuilder;
import com.azure.ai.formrecognizer.documentanalysis.administration.models.ComposeDocumentModelOptions;
import com.azure.ai.formrecognizer.documentanalysis.administration.models.DocumentModelBuildMode;
import com.azure.ai.formrecognizer.documentanalysis.administration.models.DocumentModelDetails;
import com.azure.ai.formrecognizer.documentanalysis.models.OperationResult;
import com.azure.core.credential.AzureKeyCredential;
import com.azure.core.util.polling.AsyncPollResponse;
import com.azure.core.util.polling.PollerFlux;
import java.time.Duration;
import java.util.Arrays;
import java.util.concurrent.TimeUnit;
/**
* Sample for creating a custom document analysis composed model.
* <p>
* This is useful when you have build different analysis models and want to aggregate a group of
* them into a single model that you (or a user) could use to analyze a custom document. When doing
* so, you can let the service decide which model more accurately represents the document to
* analyze, instead of manually trying each built model against the form and selecting
* the most accurate one.
* </p>
*/
public class ComposeDocumentModelAsync {
/**
* Main method to invoke this demo.
*
* @param args Unused. Arguments to the program.
*/
public static void main(final String[] args) {
// Instantiate a client that will be used to call the service.
DocumentModelAdministrationAsyncClient client = new DocumentModelAdministrationClientBuilder()
.credential(new AzureKeyCredential("{key}"))
.endpoint("https://{endpoint}.cognitiveservices.azure.com/")
.buildAsyncClient();
// Build custom document analysis model
String model1TrainingFiles = "{SAS_URL_of_your_container_in_blob_storage_for_model_1}";
// The shared access signature (SAS) Url of your Azure Blob Storage container with your forms.
PollerFlux<OperationResult, DocumentModelDetails> model1Poller =
client.beginBuildDocumentModel(model1TrainingFiles, DocumentModelBuildMode.TEMPLATE);
// Build custom document analysis model
String model2TrainingFiles = "{SAS_URL_of_your_container_in_blob_storage_for_model_2}";
// The shared access signature (SAS) Url of your Azure Blob Storage container with your forms.
PollerFlux<OperationResult, DocumentModelDetails> model2Poller =
client.beginBuildDocumentModel(model2TrainingFiles, DocumentModelBuildMode.TEMPLATE);
String labeledModelId1 = model1Poller.getSyncPoller().getFinalResult().getModelId();
String labeledModelId2 = model2Poller.getSyncPoller().getFinalResult().getModelId();
client.beginComposeDocumentModel(Arrays.asList(labeledModelId1, labeledModelId2),
new ComposeDocumentModelOptions().setDescription("my composed model desc"))
.setPollInterval(Duration.ofSeconds(5))
.flatMap(AsyncPollResponse::getFinalResult)
.subscribe(documentModel -> {
System.out.printf("Model ID: %s%n", documentModel.getModelId());
System.out.printf("Model Description: %s%n", documentModel.getDescription());
System.out.printf("Composed model created on: %s%n", documentModel.getCreatedOn());
System.out.println("Document Fields:");
documentModel.getDocumentTypes().forEach((key, documentTypeDetails) -> {
documentTypeDetails.getFieldSchema().forEach((field, documentFieldSchema) -> {
System.out.printf("Field: %s", field);
System.out.printf("Field type: %s", documentFieldSchema.getType());
System.out.printf("Field confidence: %.2f", documentTypeDetails.getFieldConfidence().get(field));
});
});
});
// The .subscribe() creation and assignment is not a blocking call. For the purpose of this example, we sleep
// the thread so the program does not end before the send operation is complete. Using .block() instead of
// .subscribe() will turn this into a synchronous call.
try {
TimeUnit.MINUTES.sleep(1);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}