• Home
• org.deeplearning4j
• deeplearning4j-scaleout-akka-word2vec
• 0.0.3.1
• source code
• CounterUtil.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.word2vec.viterbi.CounterUtil Maven / Gradle / Ivy

package org.deeplearning4j.word2vec.viterbi;

import java.util.*;

import org.deeplearning4j.berkeley.CounterMap;
import org.deeplearning4j.berkeley.Pair;
import org.jblas.DoubleMatrix;


public class CounterUtil {
	
	public static DoubleMatrix convert(CounterMap counter) {
		DoubleMatrix ret = new DoubleMatrix(counter.size(),counter.keySet().size());
		Iterator> iter = counter.getPairIterator();
		
		for(Pair next = iter.next(); iter.hasNext(); next = iter.next()) {
			int firstInt = next.getFirst();
			int secondInt = next.getSecond();
			ret.put(firstInt, secondInt,counter.getCount(next.getFirst(),next.getSecond()));
		}
		
		return ret;
	}

}