use of com.amplifyframework.predictions.models.KeyPhrase in project amplify-android by aws-amplify.
the class AWSComprehendService method comprehend.
void comprehend(@NonNull String text, @NonNull Consumer<InterpretResult> onSuccess, @NonNull Consumer<PredictionsException> onError) {
try {
// First obtain the dominant language to begin analysis
final Language dominantLanguage = fetchPredominantLanguage(text);
final LanguageType language = dominantLanguage.getValue();
// Actually analyze text in the context of dominant language
final Sentiment sentiment = fetchSentiment(text, language);
final List<KeyPhrase> keyPhrases = fetchKeyPhrases(text, language);
final List<Entity> entities = fetchEntities(text, language);
final List<Syntax> syntax = fetchSyntax(text, language);
onSuccess.accept(InterpretResult.builder().language(dominantLanguage).sentiment(sentiment).keyPhrases(keyPhrases).entities(entities).syntax(syntax).build());
} catch (PredictionsException exception) {
onError.accept(exception);
}
}
use of com.amplifyframework.predictions.models.KeyPhrase in project amplify-android by aws-amplify.
the class AWSComprehendService method fetchKeyPhrases.
private List<KeyPhrase> fetchKeyPhrases(String text, LanguageType language) throws PredictionsException {
// Skip if configuration specifies NOT key phrase
if (!isResourceConfigured(InterpretTextConfiguration.InterpretType.KEY_PHRASES)) {
return null;
}
DetectKeyPhrasesRequest request = new DetectKeyPhrasesRequest().withText(text).withLanguageCode(language.getLanguageCode());
// Detect key phrases from given text via AWS Comprehend
final DetectKeyPhrasesResult result;
try {
result = comprehend.detectKeyPhrases(request);
} catch (AmazonClientException serviceException) {
throw new PredictionsException("AWS Comprehend encountered an error while detecting key phrases.", serviceException, "See attached service exception for more details.");
}
// Convert AWS Comprehend's detection result to Amplify-compatible format
List<KeyPhrase> keyPhrases = new ArrayList<>();
for (com.amazonaws.services.comprehend.model.KeyPhrase comprehendKeyPhrase : result.getKeyPhrases()) {
KeyPhrase amplifyKeyPhrase = KeyPhrase.builder().value(comprehendKeyPhrase.getText()).confidence(comprehendKeyPhrase.getScore() * PERCENT).targetText(comprehendKeyPhrase.getText()).startIndex(comprehendKeyPhrase.getBeginOffset()).build();
keyPhrases.add(amplifyKeyPhrase);
}
return keyPhrases;
}
use of com.amplifyframework.predictions.models.KeyPhrase in project amplify-android by aws-amplify.
the class AWSPredictionsInterpretTest method testKeyPhraseDetection.
/**
* Assert that key phrases are detected.
* @throws Exception if prediction fails
*/
@Test
public void testKeyPhraseDetection() throws Exception {
final String sampleText = "My mama always said life was like a box of chocolates.";
// Interpret sample text and assert non-null result
InterpretResult result = predictions.interpret(sampleText);
assertNotNull(result);
// Assert key phrase detection
List<KeyPhrase> actual = result.getKeyPhrases();
List<String> expected = Arrays.asList("My mama", "life", "a box", "chocolates");
FeatureAssert.assertMatches(expected, actual);
}
Aggregations