2017-01-26 1 views
2

これは私が何をしたいのか、私のarffファイルStringToWordVector(weka)をjavaで使用する方法は?

@relation hamspam 

@attribute text string 
@attribute class {ham,spam} 

@data 
'good',ham 
'very good',ham 
'bad',spam 
'very bad',spam 
'very bad, very bad',spam 

は私のJavaプログラムでWEKAのclasiffierでそれを分類することですが、私はStringToWordVectorを使用し、それを分類する方法がわかりません。

この私のコードは:

Classifier j48tree = new J48();  
Instances train = new Instances(new BufferedReader(new FileReader("data.arff"))); 

StringToWordVector filter = new StringToWordVector(); 

次の?私は...

答えて

2
import weka.core.Instance; 
//import required classes 
import weka.core.Instances; 
import weka.core.converters.ConverterUtils.DataSource; 
import weka.core.stemmers.LovinsStemmer; 
import weka.classifiers.meta.FilteredClassifier; 
import weka.classifiers.trees.J48; 
import weka.filters.unsupervised.attribute.Remove; 
import weka.filters.unsupervised.attribute.StringToWordVector; 

public class ClassifierWithFilter{ 

    public static void main(String args[]) throws Exception{ 
    //load dataset 
    DataSource source = new DataSource("/Users/amaryadav/Desktop/spamham.arff"); 
    Instances dataset = source.getDataSet(); 
    //set class index to the last attribute 
    dataset.setClassIndex(dataset.numAttributes()-1); 

    //the base classifier 
    J48 tree = new J48(); 

    //the filter 
    StringToWordVector filter = new StringToWordVector(); 
    filter.setInputFormat(dataset); 
    filter.setIDFTransform(true); 
    filter.setUseStoplist(true); 
    LovinsStemmer stemmer = new LovinsStemmer(); 
    filter.setStemmer(stemmer); 
    filter.setLowerCaseTokens(true); 

    //Create the FilteredClassifier object 
    FilteredClassifier fc = new FilteredClassifier(); 
    //specify filter 
    fc.setFilter(filter); 
    //specify base classifier 
    fc.setClassifier(tree); 
    //Build the meta-classifier 
    fc.buildClassifier(dataset); 

    System.out.println(tree.graph()); 
    System.out.println(tree); 
    } 
} 

このコードはspamhamで訓練された分類器を構築するためにJ48の決定木を使用して何をするか分からない何か。 arff。希望が役立ちます。

関連する問題