Example usage for org.deeplearning4j.models.embeddings.learning.impl.elements CBOW CBOW

Introduction

In this page you can find the example usage for org.deeplearning4j.models.embeddings.learning.impl.elements CBOW CBOW.

Prototype

CBOW

Source Link

Usage

From source file:org.knime.ext.textprocessing.dl4j.nodes.embeddings.learn.w2v.Word2VecLearnerNodeModel.java

License:Open Source License

/**
 * Parse the string representation of an
 * {@link org.knime.ext.textprocessing.dl4j.settings.enumerate.ElementsLearningAlgorithm} and returns the
 * corresponding DL4J object.//from w ww  .java 2s  .c  o  m
 *
 * @param rep
 * @return DL4J object of elements algo corresponding to specified string representation
 */
private ElementsLearningAlgorithm<VocabWord> parseElementsAlgo(final String rep) {
    switch (org.knime.ext.textprocessing.dl4j.settings.enumerate.ElementsLearningAlgorithm.valueOf(rep)) {
    case CBOW:
        return new CBOW<VocabWord>();
    case SKIP_GRAM:
        return new SkipGram<VocabWord>();
    default:
        throw new IllegalArgumentException("No case defined for ElementsLearningAlgorithm: " + rep);
    }
}

---