• Home
• org.deeplearning4j
• deeplearning4j-scaleout-akka-word2vec
• 0.0.3.1
• source code
• Word2VecTrainer.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.deeplearning4j.apps.Word2VecTrainer Maven / Gradle / Ivy

package org.deeplearning4j.apps;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;

import org.deeplearning4j.word2vec.Word2Vec;
import org.deeplearning4j.word2vec.sentenceiterator.FileSentenceIterator;


public class Word2VecTrainer {

	/**
	 * @param args
	 */
	public static void main(String[] args) throws Exception {
		FileSentenceIterator f = new FileSentenceIterator(new File(args[0]));
		Word2Vec vec = new Word2Vec(f);
		vec.setMinWordFrequency(1);
		vec.train();
		BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(new File(args[1])));
		vec.write(bos);
		bos.flush();
		bos.close();
		
	}

}