org.encog.workbench.tabs.visualize.weights.AnalyzeWeightsTab.java Source code

Introduction

Here is the source code for org.encog.workbench.tabs.visualize.weights.AnalyzeWeightsTab.java

Source

/*
 * Encog(tm) Workbench v3.0
 * http://www.heatonresearch.com/encog/
 * http://code.google.com/p/encog-java/
     
 * Copyright 2008-2011 Heaton Research, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *   
 * For more information on Heaton Research copyrights, licenses 
 * and trademarks visit:
 * http://www.heatonresearch.com/copyright
 */
package org.encog.workbench.tabs.visualize.weights;

import java.awt.BorderLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;

import javax.swing.JButton;
import javax.swing.JPanel;
import javax.swing.JTabbedPane;

import org.encog.neural.networks.BasicNetwork;
import org.encog.neural.networks.structure.AnalyzeNetwork;
import org.encog.workbench.WorkBenchError;
import org.encog.workbench.frames.document.tree.ProjectEGFile;
import org.encog.workbench.frames.document.tree.ProjectFile;
import org.encog.workbench.tabs.EncogCommonTab;
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.statistics.HistogramDataset;
import org.jfree.data.xy.IntervalXYDataset;

public class AnalyzeWeightsTab extends EncogCommonTab implements ActionListener {

    /**
     * 
     */
    private static final long serialVersionUID = 4472644832610364833L;

    private JButton buttonClose;
    private JTabbedPane tabs;
    private WeightInfo weightInfo;
    private AnalyzeNetwork analyze;
    private BasicNetwork network;
    private HistogramDataset dataAll;
    private HistogramDataset dataWeights;
    private HistogramDataset dataThresholds;

    public AnalyzeWeightsTab(ProjectEGFile encogObject) {
        super(encogObject);

        if (!(encogObject.getObject() instanceof BasicNetwork)) {
            throw new WorkBenchError(
                    "Can't visualize weights of " + encogObject.getObject().getClass().getSimpleName());
        }

        this.network = (BasicNetwork) encogObject.getObject();
        this.analyze = new AnalyzeNetwork(this.network);

        this.setLayout(new BorderLayout());

        JPanel buttonPanel = new JPanel();
        add(buttonPanel, BorderLayout.SOUTH);

        this.buttonClose = new JButton("Close");
        buttonPanel.add(this.buttonClose);
        this.buttonClose.addActionListener(this);

        this.weightInfo = new WeightInfo(this);
        this.add(this.weightInfo, BorderLayout.NORTH);

        //
        createAllDataset();
        ChartPanel allChart = new ChartPanel(createChart(this.dataAll));
        ChartPanel weightChart = new ChartPanel(createChart(this.dataWeights));
        ChartPanel thresholdChart = new ChartPanel(createChart(this.dataThresholds));

        this.tabs = new JTabbedPane();
        this.add(tabs, BorderLayout.CENTER);
        this.tabs.addTab("Weights & Thresholds", allChart);
        this.tabs.addTab("Thresholds", weightChart);
        this.tabs.addTab("Weights", thresholdChart);

    }

    private void createAllDataset() {

        // all values
        this.dataAll = new HistogramDataset();
        double[] values = this.analyze.getAllValues();
        this.dataAll.addSeries("Weights & Thresholds", values, 50);

        // weight values
        this.dataThresholds = new HistogramDataset();
        double[] values2 = this.analyze.getWeightValues();
        this.dataThresholds.addSeries("Weights", values2, 50);

        // threshold values
        this.dataWeights = new HistogramDataset();
        double[] values3 = this.analyze.getBiasValues();
        this.dataWeights.addSeries("Thresholds", values3, 50);

    }

    /**
     * Creates a chart.
     * 
     * @param dataset  a dataset.
     * 
     * @return The chart.
     */
    private JFreeChart createChart(IntervalXYDataset dataset) {
        JFreeChart chart = ChartFactory.createHistogram(null, null, null, dataset, PlotOrientation.VERTICAL, true,
                false, false);
        chart.getXYPlot().setForegroundAlpha(0.75f);
        return chart;
    }

    public AnalyzeNetwork getAnalyze() {
        return this.analyze;
    }

    public void actionPerformed(ActionEvent e) {
        if (e.getSource() == this.buttonClose) {
            this.dispose();
        }

    }

    @Override
    public String getName() {
        return "Visualize :" + this.getEncogObject().getName();
    }

}