|
Java example source code file (VectorsConfigurationTest.java)
The VectorsConfigurationTest.java Java example source codepackage org.deeplearning4j.models.embeddings.loader; import org.canova.api.util.ClassPathResource; import org.deeplearning4j.models.word2vec.Word2Vec; import org.deeplearning4j.text.sentenceiterator.SentenceIterator; import org.deeplearning4j.text.sentenceiterator.UimaSentenceIterator; import org.junit.Before; import org.junit.Test; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.io.File; import static org.junit.Assert.*; /** * Created by fartovii on 21.11.15. */ public class VectorsConfigurationTest { protected static final Logger log = LoggerFactory.getLogger(VectorsConfigurationTest.class); @Before public void setUp() throws Exception { } @Test public void testFromJson() throws Exception { VectorsConfiguration configuration = new VectorsConfiguration(); configuration.setHugeModelExpected(true); configuration.setWindow(5); configuration.setIterations(3); configuration.setLayersSize(200); configuration.setLearningRate(1.4d); configuration.setSampling(0.0005d); configuration.setMinLearningRate(0.25d); configuration.setEpochs(1); String json = configuration.toJson(); log.info("Conf. JSON: " + json); VectorsConfiguration configuration2 = VectorsConfiguration.fromJson(json); assertEquals(configuration, configuration2); } @Test public void testFromW2V() throws Exception { VectorsConfiguration configuration = new VectorsConfiguration(); configuration.setHugeModelExpected(true); configuration.setWindow(5); configuration.setIterations(3); configuration.setLayersSize(200); configuration.setLearningRate(1.4d); configuration.setSampling(0.0005d); configuration.setMinLearningRate(0.25d); configuration.setEpochs(1); File inputFile = new ClassPathResource("/big/raw_sentences.txt").getFile(); SentenceIterator iter = UimaSentenceIterator.createWithPath(inputFile.getAbsolutePath()); Word2Vec vec = new Word2Vec.Builder(configuration) .iterate(iter) .build(); VectorsConfiguration configuration2 = vec.getConfiguration(); assertEquals(configuration, configuration2); } } Other Java examples (source code examples)Here is a short list of links related to this Java VectorsConfigurationTest.java source code file: |
... this post is sponsored by my books ... | |
#1 New Release! |
FP Best Seller |
Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.
A percentage of advertising revenue from
pages under the /java/jwarehouse
URI on this website is
paid back to open source projects.