From 5d54051367c289577b2ff270fd431f0004a3a889 Mon Sep 17 00:00:00 2001 From: ellerenad Date: Sun, 19 Jul 2020 23:14:04 +0200 Subject: [PATCH] export trained Word2Vec model --- .../embeddingsfromcorpus/word2vec/Word2VecRawTextExample.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dl4j-examples/src/main/java/org/deeplearning4j/examples/advanced/modelling/embeddingsfromcorpus/word2vec/Word2VecRawTextExample.java b/dl4j-examples/src/main/java/org/deeplearning4j/examples/advanced/modelling/embeddingsfromcorpus/word2vec/Word2VecRawTextExample.java index 86125eed4b..94fe521a69 100644 --- a/dl4j-examples/src/main/java/org/deeplearning4j/examples/advanced/modelling/embeddingsfromcorpus/word2vec/Word2VecRawTextExample.java +++ b/dl4j-examples/src/main/java/org/deeplearning4j/examples/advanced/modelling/embeddingsfromcorpus/word2vec/Word2VecRawTextExample.java @@ -17,6 +17,7 @@ package org.deeplearning4j.examples.advanced.modelling.embeddingsfromcorpus.word2vec; +import org.deeplearning4j.models.embeddings.loader.WordVectorSerializer; import org.deeplearning4j.models.word2vec.Word2Vec; import org.deeplearning4j.examples.utils.DownloaderUtility; import org.deeplearning4j.text.sentenceiterator.BasicLineIterator; @@ -77,7 +78,8 @@ public static void main(String[] args) throws Exception { vec.fit(); log.info("Writing word vectors to text file...."); - + WordVectorSerializer.writeWord2VecModel(vec, "pathToSaveModel.txt"); + // Prints out the closest 10 words to "day". An example on what to do with these Word Vectors. log.info("Closest Words:"); Collection lst = vec.wordsNearestSum("day", 10);