use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.
the class SpeechToTextIT method testListWordsTypeAll.
/**
* Test list words with type all.
*/
@Test
@Ignore
public void testListWordsTypeAll() {
ListWordsOptions listOptions = new ListWordsOptions.Builder().customizationId(customizationId).wordType(ListWordsOptions.WordType.ALL).build();
Words result = service.listWords(listOptions).execute();
assertNotNull(result);
assertTrue(!result.getWords().isEmpty());
}
use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.
the class SpeechToTextIT method testCreateLanguageModel.
/**
* Test create language model.
*
* @throws InterruptedException the interrupted exception
*/
@Test
public void testCreateLanguageModel() throws InterruptedException, FileNotFoundException {
CreateLanguageModelOptions createOptions = new CreateLanguageModelOptions.Builder().name("java-sdk-temporary").baseModelName(EN_BROADBAND16K).description("Temporary custom model for testing the Java SDK").build();
LanguageModel myModel = service.createLanguageModel(createOptions).execute();
String id = myModel.getCustomizationId();
try {
// Add a corpus file to the model
AddCorpusOptions addOptions = new AddCorpusOptions.Builder().customizationId(id).corpusName("corpus-1").corpusFile(new File(String.format(SPEECH_RESOURCE, "corpus1.txt"))).corpusFileContentType(HttpMediaType.TEXT_PLAIN).allowOverwrite(false).build();
service.addCorpus(addOptions).execute();
// Get corpus status
GetCorpusOptions getOptions = new GetCorpusOptions.Builder().customizationId(id).corpusName("corpus-1").build();
for (int x = 0; x < 30 && !service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED); x++) {
Thread.sleep(5000);
}
assertTrue(service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED));
// Add the corpus file to the model again and allow overwrite
AddCorpusOptions addOptionsWithOverwrite = new AddCorpusOptions.Builder().customizationId(id).corpusName("corpus-1").corpusFile(new File(String.format(SPEECH_RESOURCE, "corpus1.txt"))).corpusFileContentType(HttpMediaType.TEXT_PLAIN).allowOverwrite(true).build();
service.addCorpus(addOptionsWithOverwrite).execute();
// Get corpus status
for (int x = 0; x < 30 && !service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED); x++) {
Thread.sleep(5000);
}
assertTrue(service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED));
// Get corpora
ListCorporaOptions listCorporaOptions = new ListCorporaOptions.Builder().customizationId(id).build();
Corpora corpora = service.listCorpora(listCorporaOptions).execute();
assertNotNull(corpora);
assertTrue(corpora.getCorpora().size() == 1);
// Now add some user words to the custom model
service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("IEEE").word("IEEE").displayAs("IEEE").addSoundsLike("I. triple E.").build()).execute();
service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("hhonors").word("hhonors").displayAs("IEEE").addSoundsLike("H. honors").addSoundsLike("Hilton honors").build()).execute();
service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("aaa").word("aaa").displayAs("aaa").addSoundsLike("aaa").addSoundsLike("bbb").build()).execute();
service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("bbb").word("bbb").addSoundsLike("aaa").addSoundsLike("bbb").build()).execute();
service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("ccc").word("ccc").displayAs("ccc").build()).execute();
service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("ddd").word("ddd").build()).execute();
service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("eee").word("eee").build()).execute();
// Display all words in the words resource (coming from OOVs from the corpus add and the new words just added)
ListWordsOptions listWordsOptions = new ListWordsOptions.Builder().customizationId(id).wordType(ListWordsOptions.WordType.ALL).build();
Words words = service.listWords(listWordsOptions).execute();
assertNotNull(words);
} finally {
DeleteLanguageModelOptions deleteOptions = new DeleteLanguageModelOptions.Builder().customizationId(id).build();
service.deleteLanguageModel(deleteOptions).execute();
}
}
use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.
the class SpeechToTextIT method testListWordsIdAndType.
/**
* Test list words with a customization ID and word type.
*/
@Test
@Ignore
public void testListWordsIdAndType() {
ListWordsOptions listOptions = new ListWordsOptions.Builder().customizationId(customizationId).wordType(ListWordsOptions.WordType.CORPORA).build();
Words result = service.listWords(listOptions).execute();
assertNotNull(result);
assertTrue(!result.getWords().isEmpty());
}
use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.
the class SpeechToTextTest method testAddWords.
/**
* Test add words.
*
* @throws InterruptedException the interrupted exception
* @throws FileNotFoundException the file not found exception
*/
@Test
public void testAddWords() throws InterruptedException, FileNotFoundException {
String id = "foo";
Word newWord = loadFixture("src/test/resources/speech_to_text/word.json", Word.class);
Map<String, Object> wordsAsMap = new HashMap<String, Object>();
wordsAsMap.put("words", new Word[] { newWord });
server.enqueue(new MockResponse().addHeader(CONTENT_TYPE, HttpMediaType.APPLICATION_JSON).setBody("{}"));
CustomWord word = new CustomWord();
word.setWord(newWord.getWord());
word.setDisplayAs(newWord.getDisplayAs());
word.setSoundsLike(newWord.getSoundsLike());
AddWordsOptions addOptions = new AddWordsOptions.Builder().customizationId(id).addWords(word).build();
service.addWords(addOptions).execute();
final RecordedRequest request = server.takeRequest();
assertEquals("POST", request.getMethod());
assertEquals(String.format(PATH_WORDS, id), request.getPath());
assertEquals(GSON.toJson(wordsAsMap), request.getBody().readUtf8());
}
use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.
the class SpeechToTextTest method testListWordsType.
/**
* Test list words with word type all.
*
* @throws InterruptedException the interrupted exception
* @throws FileNotFoundException the file not found exception
*/
@Test
public void testListWordsType() throws InterruptedException, FileNotFoundException {
String id = "foo";
String wordsAsStr = getStringFromInputStream(new FileInputStream("src/test/resources/speech_to_text/words.json"));
JsonObject words = new JsonParser().parse(wordsAsStr).getAsJsonObject();
server.enqueue(new MockResponse().addHeader(CONTENT_TYPE, HttpMediaType.APPLICATION_JSON).setBody(wordsAsStr));
ListWordsOptions listOptions = new ListWordsOptions.Builder().customizationId(id).wordType(ListWordsOptions.WordType.ALL).build();
Words result = service.listWords(listOptions).execute();
final RecordedRequest request = server.takeRequest();
assertEquals("GET", request.getMethod());
assertEquals(String.format(PATH_WORDS, id) + "?word_type=all", request.getPath());
assertEquals(words.get("words"), GSON.toJsonTree(result.getWords()));
}
Aggregations