use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Word 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.Word in project java-sdk by watson-developer-cloud.
the class SpeechToTextIT method testGetWord.
/**
* Test get word.
*/
public void testGetWord() {
GetWordOptions getOptions = new GetWordOptions.Builder().customizationId(customizationId).wordName("string").build();
Word result = service.getWord(getOptions).execute();
assertNotNull(result);
}
use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Word 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.Word in project java-sdk by watson-developer-cloud.
the class SpeechToTextTest method testAddWord.
/**
* Test add word.
*
* @throws InterruptedException the interrupted exception
* @throws FileNotFoundException the file not found exception
*/
@Test
public void testAddWord() throws InterruptedException, FileNotFoundException {
String id = "foo";
Word newWord = loadFixture("src/test/resources/speech_to_text/word.json", Word.class);
server.enqueue(new MockResponse().addHeader(CONTENT_TYPE, HttpMediaType.APPLICATION_JSON).setBody("{}"));
AddWordOptions addOptions = new AddWordOptions.Builder().wordName(newWord.getWord()).customizationId(id).word(newWord.getWord()).displayAs(newWord.getDisplayAs()).soundsLike(newWord.getSoundsLike()).build();
service.addWord(addOptions).execute();
final RecordedRequest request = server.takeRequest();
assertEquals("PUT", request.getMethod());
assertEquals(String.format(PATH_WORD, id, newWord.getWord()), request.getPath());
assertEquals(GSON.toJson(newWord), request.getBody().readUtf8());
}
use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Word in project java-sdk by watson-developer-cloud.
the class SpeechToTextTest method testDeleteWord.
/**
* Test delete word.
*
* @throws InterruptedException the interrupted exception
*/
@Test
public void testDeleteWord() throws InterruptedException {
String id = "foo";
String wordName = "bar";
server.enqueue(new MockResponse().addHeader(CONTENT_TYPE, HttpMediaType.APPLICATION_JSON).setBody("{}"));
DeleteWordOptions deleteOptions = new DeleteWordOptions.Builder().customizationId(id).wordName(wordName).build();
service.deleteWord(deleteOptions).execute();
final RecordedRequest request = server.takeRequest();
assertEquals("DELETE", request.getMethod());
assertEquals(String.format(PATH_WORD, id, wordName), request.getPath());
}
Aggregations