Package weka.classifiers.evaluation

Class EvaluationUtils

  • All Implemented Interfaces:
    RevisionHandler
    public class EvaluationUtils
extends java.lang.Object
implements RevisionHandler
    Contains utility functions for generating lists of predictions in various manners.
    Version:
    $Revision: 1.11 $
    Author:
    Len Trigg (len@reeltwo.com)

    • Constructor Detail

      • EvaluationUtils

        public EvaluationUtils()

    • Method Detail

      • 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 FastVector getCVPredictions​(Classifier classifier,
                                   Instances data,
                                   int numFolds)
                            throws java.lang.Exception
        Generate a bunch of predictions ready for processing, by performing a cross-validation on the supplied dataset.
        Parameters:
        classifier - the Classifier to evaluate
        data - the dataset
        numFolds - the number of folds in the cross-validation.
        Throws:
        java.lang.Exception - if an error occurs

      • getTrainTestPredictions

        public FastVector getTrainTestPredictions​(Classifier classifier,
                                          Instances train,
                                          Instances test)
                                   throws java.lang.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 evaluate
        train - the training dataset
        test - the test dataset
        Throws:
        java.lang.Exception - if an error occurs

      • getTestPredictions

        public FastVector getTestPredictions​(Classifier classifier,
                                     Instances test)
                              throws java.lang.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 evaluate
        test - the test dataset
        Throws:
        java.lang.Exception - if an error occurs

      • getPrediction

        public Prediction getPrediction​(Classifier classifier,
                                Instance test)
                         throws java.lang.Exception
        Generate a single prediction for a test instance given the pre-trained classifier.
        Parameters:
        classifier - the pre-trained Classifier to evaluate
        test - the test instance
        Throws:
        java.lang.Exception - if an error occurs

      • getRevision

        public java.lang.String getRevision()
        Returns the revision string.
        Specified by:
        getRevision in interface RevisionHandler
        Returns:
        the revision