Package weka.classifiers.evaluation
Class EvaluationUtils
java.lang.Object
weka.classifiers.evaluation.EvaluationUtils
- All Implemented Interfaces:
RevisionHandler
Contains utility functions for generating lists of predictions in various
manners.
- Version:
- $Revision: 10153 $
- Author:
- Len Trigg (len@reeltwo.com)
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptiongetCVPredictions(Classifier classifier, Instances data, int numFolds) Generate a bunch of predictions ready for processing, by performing a cross-validation on the supplied dataset.getPrediction(Classifier classifier, Instance test) Generate a single prediction for a test instance given the pre-trained classifier.Returns the revision string.intgetSeed()Gets the seed for randomization during cross-validationgetTestPredictions(Classifier classifier, Instances test) Generate a bunch of predictions ready for processing, by performing a evaluation on a test set assuming the classifier is already trained.getTrainTestPredictions(Classifier classifier, Instances train, Instances test) Generate a bunch of predictions ready for processing, by performing a evaluation on a test set after training on the given training set.voidsetSeed(int seed) Sets the seed for randomization during cross-validation
-
Constructor Details
-
EvaluationUtils
public EvaluationUtils()
-
-
Method Details
-
setSeed
public void setSeed(int seed) Sets the seed for randomization during cross-validation -
getSeed
public int getSeed()Gets the seed for randomization during cross-validation -
getCVPredictions
public ArrayList<Prediction> getCVPredictions(Classifier classifier, Instances data, int numFolds) throws Exception Generate a bunch of predictions ready for processing, by performing a cross-validation on the supplied dataset.- Parameters:
classifier- the Classifier to evaluatedata- the datasetnumFolds- the number of folds in the cross-validation.- Throws:
Exception- if an error occurs
-
getTrainTestPredictions
public ArrayList<Prediction> getTrainTestPredictions(Classifier classifier, Instances train, Instances test) throws Exception Generate a bunch of predictions ready for processing, by performing a evaluation on a test set after training on the given training set.- Parameters:
classifier- the Classifier to evaluatetrain- the training datasettest- the test dataset- Throws:
Exception- if an error occurs
-
getTestPredictions
public ArrayList<Prediction> getTestPredictions(Classifier classifier, Instances test) throws Exception Generate a bunch of predictions ready for processing, by performing a evaluation on a test set assuming the classifier is already trained.- Parameters:
classifier- the pre-trained Classifier to evaluatetest- the test dataset- Throws:
Exception- if an error occurs
-
getPrediction
Generate a single prediction for a test instance given the pre-trained classifier.- Parameters:
classifier- the pre-trained Classifier to evaluatetest- the test instance- Throws:
Exception- if an error occurs
-
getRevision
Returns the revision string.- Specified by:
getRevisionin interfaceRevisionHandler- Returns:
- the revision
-