
org.deeplearning4j.word2vec.viterbi.Datum Maven / Gradle / Ivy
The newest version!
package org.deeplearning4j.word2vec.viterbi;
import java.util.*;
import org.deeplearning4j.word2vec.util.Window;
public class Datum {
public final String word;
public final String label;
public List features;
public String guessLabel;
public String previousLabel;
public Datum(String word, String label) {
this.word = word;
this.label = label;
}
public static List datums(List windows,List labels,List previousLabels,List> features) {
List ret = new ArrayList();
for(int i = 0; i < windows.size(); i++) {
ret.add(create(windows.get(i),labels.get(i),previousLabels.get(i),features.get(i)));
}
return ret;
}
public static Datum create(Window window,String label,String prevLabel,List features) {
Datum ret = new Datum(window.getFocusWord(),label);
ret.previousLabel = prevLabel;
ret.features = features;
return ret;
}
public static Datum create(Window window,String label,String prevLabel) {
Datum ret = new Datum(window.getFocusWord(),label);
ret.previousLabel = prevLabel;
return ret;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy