aurora: Model Evaluation

Evaluate your model

Predictable model = //... your model (NeuralNetwork, LinearRegression, etc.)

1. Create a TestSet to match the trained inputs & outputs

Map<double[], double[]> data = new HashMap<>();
data.put(new double[]{0.0, 0.0}, new double[]{0.0});
data.put(new double[]{0.0, 1.0}, new double[]{1.0});
data.put(new double[]{1.0, 0.0}, new double[]{1.0});
data.put(new double[]{1.0, 1.0}, new double[]{0.0});
TestSet testSet = new TestSet(data);

2. Evaluate

Evaluation eval = model.evaluate(testSet);

3. Print summary

eval.printSummary();

Prints:

Evaluation Summary of Type: (Model Type)
 Accuracy: x
 Precision: x
 Recall: x
 F1 Score: x