ai.libs.mlplan.gui.outofsampleplots.OutOfSampleErrorPlotPluginView Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mlplancli Show documentation

This project provides an implementation of the AutoML tool ML-Plan.

The newest version!

package ai.libs.mlplan.gui.outofsampleplots;

import java.util.ArrayList;
import java.util.List;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ai.libs.jaicore.graphvisualizer.plugin.ASimpleMVCPluginView;
import javafx.application.Platform;
import javafx.scene.chart.LineChart;
import javafx.scene.chart.NumberAxis;
import javafx.scene.chart.XYChart.Data;
import javafx.scene.chart.XYChart.Series;

/**
 * 
 * @author fmohr
 *
 */
public class OutOfSampleErrorPlotPluginView extends ASimpleMVCPluginView> {

	private Logger logger = LoggerFactory.getLogger(OutOfSampleErrorPlotPluginView.class);
	private final Series believedErrorSeries;
	private final Series outOfSampleErrorSeries;
	private int nextIndexToDisplay = 0;

	public OutOfSampleErrorPlotPluginView(OutOfSampleErrorPlotPluginModel model) {
		super(model, new LineChart<>(new NumberAxis(), new NumberAxis()));
		getNode().getXAxis().setLabel("elapsed time (s)");
		getNode().setTitle(getTitle());
		believedErrorSeries = new Series<>();
		believedErrorSeries.setName("Believed (internal) Error");
		outOfSampleErrorSeries = new Series<>();
		outOfSampleErrorSeries.setName("Out-of-Sample Error");
		getNode().getData().add(believedErrorSeries);
		getNode().getData().add(outOfSampleErrorSeries);
	}

	@Override
	public void update() {
		
		/* compute data to add */
		List observedTimestamps = getModel().getTimestamps();
		List> performances = getModel().getPerformances();
		List> believedErrors = new ArrayList<>();
		List> outOfSampleErrors = new ArrayList<>();
		for (; nextIndexToDisplay < observedTimestamps.size(); nextIndexToDisplay++) {
			int timestamp = observedTimestamps.get(nextIndexToDisplay) / 100;
			believedErrors.add(new Data<>(timestamp, performances.get(nextIndexToDisplay).get(0)));
			outOfSampleErrors.add(new Data<>(timestamp, performances.get(nextIndexToDisplay).get(1)));
		}
		logger.info("Adding {} values to chart.", believedErrors.size());
		Platform.runLater(() -> {
			believedErrorSeries.getData().addAll(believedErrors);
			outOfSampleErrorSeries.getData().addAll(outOfSampleErrors);
		});

	}

	@Override
	public String getTitle() {
		return "Out-of-Sample Error Timeline";
	}
	
	public void clear() {
		nextIndexToDisplay = 0;
		believedErrorSeries.getData().clear();
		outOfSampleErrorSeries.getData().clear();
	}

	public int getNextIndexToDisplay() {
		return nextIndexToDisplay;
	}	
}