Skip to content

Commit

Permalink
Fixed small error with iris classification training error report.
Browse files Browse the repository at this point in the history
  • Loading branch information
jeffheaton committed Oct 4, 2014
1 parent 0a840ca commit c059715
Showing 1 changed file with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@
import org.encog.ml.model.EncogModel;
import org.encog.util.csv.CSVFormat;
import org.encog.util.csv.ReadCSV;
import org.encog.util.simple.EncogUtility;

public class IrisClassification {
public static String DATA_URL = "https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data";
Expand Down Expand Up @@ -117,8 +118,8 @@ public void run(String[] args) {
MLRegression bestMethod = (MLRegression)model.crossvalidate(5, true);

// Display the training and validation errors.
System.out.println( "Training error: " + model.calculateError(bestMethod, model.getTrainingDataset()));
System.out.println( "Validation error: " + model.calculateError(bestMethod, model.getValidationDataset()));
System.out.println( "Training error: " + EncogUtility.calculateRegressionError(bestMethod, model.getTrainingDataset()));
System.out.println( "Validation error: " + EncogUtility.calculateRegressionError(bestMethod, model.getValidationDataset()));

// Display our normalization parameters.
NormalizationHelper helper = data.getNormHelper();
Expand Down

0 comments on commit c059715

Please sign in to comment.