テキストの語句化と量子化

1775 ワード

<strong><span style="font-size:18px;">/***
 * @author YangXin
 * @info  ,  ,
 *  。 StaticWordValueEncoder , 
 *  。Mahout ,Lucene 。
 */
package unitFourteen;

import java.io.IOException;
import java.io.StringReader;

import org.apache.commons.collections.bag.SynchronizedSortedBag;
import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.TokenStream;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.analysis.tokenattributes.TermAttribute;
import org.apache.lucene.util.Version;
import org.apache.mahout.math.RandomAccessSparseVector;
import org.apache.mahout.math.SequentialAccessSparseVector;
import org.apache.mahout.math.Vector;
import org.apache.mahout.vectorizer.encoders.FeatureVectorEncoder;
import org.apache.mahout.vectorizer.encoders.StaticWordValueEncoder;

public class TokenizingAndVectorizingText {
	public static void main(String[] args) throws IOException {
		FeatureVectorEncoder encoder = new StaticWordValueEncoder("text");
		Analyzer analyzer = new StandardAnalyzer(Version.LUCENE_31);     

		StringReader in = new StringReader("text to magically vectorize");
		TokenStream ts = analyzer.tokenStream("body", in);
		TermAttribute termAtt = ts.addAttribute(TermAttribute.class);

		Vector v1 = new RandomAccessSparseVector(100);                   
		while (ts.incrementToken()) {
		  char[] termBuffer = termAtt.termBuffer();
		  int termLen = termAtt.termLength();
		  String w = new String(termBuffer, 0, termLen);                 
		  encoder.addToVector(w, 1, v1);                                 
		}
		System.out.printf("%s
", new SequentialAccessSparseVector(v1)); } } </span></strong>