private XYDataset createDataset(String metricName) { series = new XYSeriesCollection(); // For each collection of evaluations for (int i = 0; i < evaluationsCollection.size(); i++) { if (!set.contains(i)) { List<AbstractEvaluation> evaluations = evaluationsCollection.get(i); XYSeries newXYSerie = new XYSeries(queryNames.get(i)); int evalIndex = 0; for (AbstractEvaluation eval : evaluations) { if (evalIndex % reportFrecuency == 0) { newXYSerie.add(eval.getLabeledSetSize(), eval.getMetricValue(metricName)); } ++evalIndex; } series.addSeries(newXYSerie); } } // The series that belongs to passive learning is the last if (passiveEvaluation != null) { series.addSeries(createPassiveLearningSerie(metricName, evaluationsCollection.get(0))); } return series; }
public XYSeries createPassiveLearningSerie( String metricName, List<AbstractEvaluation> evaluations) { XYSeries xyseriesPassive = new XYSeries("Passive learning"); // Fill the series of passive learning int evalIndex = 0; for (AbstractEvaluation eval : evaluations) { if (evalIndex % reportFrecuency == 0) { xyseriesPassive.add(eval.getLabeledSetSize(), passiveEvaluation.getMetricValue(metricName)); } ++evalIndex; } return xyseriesPassive; }