温馨提示×

Java ARIMA模型结果可视化方法

小樊
83
2024-08-07 06:25:26
栏目: 编程语言

在Java中,可以使用JFreeChart库来进行ARIMA模型结果的可视化。以下是一个简单的示例代码,用来将ARIMA模型拟合的结果可视化:

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.jfree.ui.ApplicationFrame;

public class ARIMAVisualization extends ApplicationFrame {

    public ARIMAVisualization(String title, double[] actualData, double[] fittedData) {
        super(title);

        XYSeries actualSeries = new XYSeries("Actual Data");
        for (int i = 0; i < actualData.length; i++) {
            actualSeries.add(i, actualData[i]);
        }

        XYSeries fittedSeries = new XYSeries("Fitted Data");
        for (int i = 0; i < fittedData.length; i++) {
            fittedSeries.add(i, fittedData[i]);
        }

        XYSeriesCollection dataset = new XYSeriesCollection();
        dataset.addSeries(actualSeries);
        dataset.addSeries(fittedSeries);

        JFreeChart chart = ChartFactory.createXYLineChart(
                "ARIMA Model Fitting Results",
                "Time",
                "Value",
                dataset,
                PlotOrientation.VERTICAL,
                true,
                true,
                false
        );

        ChartPanel chartPanel = new ChartPanel(chart);
        chartPanel.setPreferredSize(new java.awt.Dimension(800, 600));
        setContentPane(chartPanel);
    }

    public static void main(String[] args) {
        double[] actualData = {10, 20, 30, 40, 50, 60, 70, 80, 90, 100};
        double[] fittedData = {12, 22, 32, 42, 52, 62, 72, 82, 92, 102};

        ARIMAVisualization demo = new ARIMAVisualization("ARIMA Model Visualization", actualData, fittedData);
        demo.pack();
        demo.setVisible(true);
    }
}

在上面的示例代码中,我们首先创建了一个主类ARIMAVisualization,继承自ApplicationFrame,然后在构造函数中初始化了实际数据和拟合数据的XYSeries对象,并将它们添加到XYSeriesCollection对象中。接着创建了一个JFreeChart对象,并使用ChartFactory类的createXYLineChart方法创建了一个折线图,并将实际数据和拟合数据添加到图表中。最后创建了一个ChartPanel对象,并将图表添加到ChartPanel中,设置图表的尺寸并显示。

你可以根据自己的实际数据和ARIMA模型拟合结果,修改实际数据actualData和拟合数据fittedData的数值,运行代码进行可视化展示。

0