weka.gui.beans.AttributeSummarizer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of weka-dev Show documentation
Show all versions of weka-dev Show documentation
The Waikato Environment for Knowledge Analysis (WEKA), a machine
learning workbench. This version represents the developer version, the
"bleeding edge" of development, you could say. New functionality gets added
to this version.
/*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see m_plots;
/**
* Creates a new AttributeSummarizer instance.
*/
public AttributeSummarizer() {
useDefaultVisual();
m_visual.setText("AttributeSummarizer");
// java.awt.GraphicsEnvironment ge =
// java.awt.GraphicsEnvironment.getLocalGraphicsEnvironment();
if (!GraphicsEnvironment.isHeadless()) {
appearanceFinal();
}
}
/**
* Global info for this bean
*
* @return a String value
*/
@Override
public String globalInfo() {
return "Plot summary bar charts for incoming data/training/test sets.";
}
/**
* Set the coloring index for the attribute summary plots
*
* @param ci an int value
*/
public void setColoringIndex(int ci) {
m_coloringIndex = ci;
}
/**
* Return the coloring index for the attribute summary plots
*
* @return an int value
*/
public int getColoringIndex() {
return m_coloringIndex;
}
/**
* Set the width of the grid of plots
*
* @param gw the width of the grid
*/
public void setGridWidth(int gw) {
if (gw > 0) {
m_bcSupport.firePropertyChange("gridWidth", new Integer(m_gridWidth),
new Integer(gw));
m_gridWidth = gw;
}
}
/**
* Get the width of the grid of plots
*
* @return the grid width
*/
public int getGridWidth() {
return m_gridWidth;
}
/**
* Set the maximum number of plots to display
*
* @param mp the number of plots to display
*/
public void setMaxPlots(int mp) {
if (mp > 0) {
m_bcSupport.firePropertyChange("maxPlots", new Integer(m_maxPlots),
new Integer(mp));
m_maxPlots = mp;
}
}
/**
* Get the number of plots to display
*
* @return the number of plots to display
*/
public int getMaxPlots() {
return m_maxPlots;
}
/**
* Set whether the appearance of this bean should be design or application
*
* @param design true if bean should appear in design mode
*/
public void setDesign(boolean design) {
m_design = true;
appearanceDesign();
}
@Override
protected void appearanceDesign() {
removeAll();
setLayout(new BorderLayout());
add(m_visual, BorderLayout.CENTER);
}
@Override
protected void appearanceFinal() {
removeAll();
setLayout(new BorderLayout());
}
@Override
protected void setUpFinal() {
removeAll();
if (m_visualizeDataSet == null) {
return;
}
if (!m_runningAsPerspective || m_activePerspective) {
final JScrollPane hp = makePanel();
add(hp, BorderLayout.CENTER);
if (m_showClassCombo) {
Vector atts = new Vector();
for (int i = 0; i < m_visualizeDataSet.numAttributes(); i++) {
atts.add("("
+ Attribute.typeToStringShort(m_visualizeDataSet.attribute(i))
+ ") " + m_visualizeDataSet.attribute(i).name());
}
final JComboBox classCombo = new JComboBox();
classCombo.setModel(new DefaultComboBoxModel(atts));
if (atts.size() > 0) {
if (m_visualizeDataSet.classIndex() < 0) {
classCombo.setSelectedIndex(atts.size() - 1);
} else {
classCombo.setSelectedIndex(m_visualizeDataSet.classIndex());
}
classCombo.setEnabled(true);
for (int i = 0; i < m_plots.size(); i++) {
m_plots.get(i).setColoringIndex(classCombo.getSelectedIndex());
}
}
JPanel comboHolder = new JPanel();
comboHolder.setLayout(new BorderLayout());
JPanel tempHolder = new JPanel();
tempHolder.setLayout(new BorderLayout());
tempHolder.add(new JLabel("Class: "), BorderLayout.WEST);
tempHolder.add(classCombo, BorderLayout.EAST);
comboHolder.add(tempHolder, BorderLayout.WEST);
add(comboHolder, BorderLayout.NORTH);
classCombo.addActionListener(new ActionListener() {
@Override
public void actionPerformed(ActionEvent e) {
int selected = classCombo.getSelectedIndex();
if (selected >= 0) {
for (int i = 0; i < m_plots.size(); i++) {
m_plots.get(i).setColoringIndex(selected);
}
}
}
});
}
}
}
/**
* Use the default appearance for this bean
*/
@Override
public void useDefaultVisual() {
m_visual.loadIcons(BeanVisual.ICON_PATH + "AttributeSummarizer.gif",
BeanVisual.ICON_PATH + "AttributeSummarizer_animated.gif");
}
/**
* Return an enumeration of actions that the user can ask this bean to perform
*
* @return an Enumeration value
*/
@Override
public Enumeration enumerateRequests() {
Vector newVector = new Vector(0);
if (m_visualizeDataSet != null) {
newVector.addElement("Show summaries");
}
return newVector.elements();
}
private JScrollPane makePanel() {
String fontFamily = this.getFont().getFamily();
Font newFont = new Font(fontFamily, Font.PLAIN, 10);
JPanel hp = new JPanel();
hp.setFont(newFont);
int numPlots = Math.min(m_visualizeDataSet.numAttributes(), m_maxPlots);
int gridHeight = numPlots / m_gridWidth;
if (numPlots % m_gridWidth != 0) {
gridHeight++;
}
hp.setLayout(new GridLayout(gridHeight, 4));
m_plots = new ArrayList();
for (int i = 0; i < numPlots; i++) {
JPanel temp = new JPanel();
temp.setLayout(new BorderLayout());
temp.setBorder(BorderFactory.createTitledBorder(m_visualizeDataSet
.attribute(i).name()));
AttributeVisualizationPanel ap = new AttributeVisualizationPanel();
m_plots.add(ap);
ap.setInstances(m_visualizeDataSet);
if (m_coloringIndex < 0 && m_visualizeDataSet.classIndex() >= 0) {
ap.setColoringIndex(m_visualizeDataSet.classIndex());
} else {
ap.setColoringIndex(m_coloringIndex);
}
temp.add(ap, BorderLayout.CENTER);
ap.setAttribute(i);
hp.add(temp);
}
Dimension d = new Dimension(830, gridHeight * 100);
hp.setMinimumSize(d);
hp.setMaximumSize(d);
hp.setPreferredSize(d);
JScrollPane scroller = new JScrollPane(hp);
return scroller;
}
/**
* Set a bean context for this bean
*
* @param bc a BeanContext value
*/
/*
* public void setBeanContext(BeanContext bc) { m_beanContext = bc; m_design =
* m_beanContext.isDesignTime(); if (m_design) { appearanceDesign(); } }
*/
/**
* Set instances for this bean. This method is a convenience method for
* clients who use this component programatically
*
* @param inst an Instances value
* @exception Exception if an error occurs
*/
@Override
public void setInstances(Instances inst) throws Exception {
if (m_design) {
throw new Exception("This method is not to be used during design "
+ "time. It is meant to be used if this "
+ "bean is being used programatically as as "
+ "stand alone component.");
}
m_visualizeDataSet = inst;
setUpFinal();
}
/**
* Returns true if this perspective accepts instances
*
* @return true if this perspective can accept instances
*/
@Override
public boolean acceptsInstances() {
return true;
}
/**
* Get the title of this perspective
*
* @return the title of this perspective
*/
@Override
public String getPerspectiveTitle() {
return "Attribute summary";
}
/**
* Get the tool tip text for this perspective.
*
* @return the tool tip text for this perspective
*/
@Override
public String getPerspectiveTipText() {
return "Matrix of attribute summary histograms";
}
/**
* Get the icon for this perspective.
*
* @return the Icon for this perspective (or null if the perspective does not
* have an icon)
*/
@Override
public Icon getPerspectiveIcon() {
java.awt.Image pic = null;
java.net.URL imageURL = this.getClass().getClassLoader()
.getResource("weka/gui/beans/icons/chart_bar.png");
if (imageURL == null) {
} else {
pic = java.awt.Toolkit.getDefaultToolkit().getImage(imageURL);
}
return new javax.swing.ImageIcon(pic);
}
/**
* Set active status of this perspective. True indicates that this perspective
* is the visible active perspective in the KnowledgeFlow
*
* @param active true if this perspective is the active one
*/
@Override
public void setActive(boolean active) {
m_activePerspective = active;
m_plots = null;
setUpFinal();
}
/**
* Set whether this perspective is "loaded" - i.e. whether or not the user has
* opted to have it available in the perspective toolbar. The perspective can
* make the decision as to allocating or freeing resources on the basis of
* this.
*
* @param loaded true if the perspective is available in the perspective
* toolbar of the KnowledgeFlow
*/
@Override
public void setLoaded(boolean loaded) {
}
/**
* Set a reference to the main KnowledgeFlow perspective - i.e. the
* perspective that manages flow layouts.
*
* @param main the main KnowledgeFlow perspective.
*/
@Override
public void setMainKFPerspective(KnowledgeFlowApp.MainKFPerspective main) {
m_showClassCombo = true;
m_runningAsPerspective = true;
}
/**
* Perform a named user request
*
* @param request a string containing the name of the request to perform
* @exception IllegalArgumentException if request is not supported
*/
@Override
public void performRequest(String request) {
if (m_design == false) {
setUpFinal();
return;
}
if (request.compareTo("Show summaries") == 0) {
try {
// popup matrix panel
if (!m_framePoppedUp) {
m_framePoppedUp = true;
final JScrollPane holderP = makePanel();
final javax.swing.JFrame jf = new javax.swing.JFrame("Visualize");
jf.setSize(800, 600);
jf.getContentPane().setLayout(new BorderLayout());
jf.getContentPane().add(holderP, BorderLayout.CENTER);
jf.addWindowListener(new java.awt.event.WindowAdapter() {
@Override
public void windowClosing(java.awt.event.WindowEvent e) {
jf.dispose();
m_framePoppedUp = false;
}
});
jf.setVisible(true);
m_popupFrame = jf;
} else {
m_popupFrame.toFront();
}
} catch (Exception ex) {
ex.printStackTrace();
m_framePoppedUp = false;
}
} else {
throw new IllegalArgumentException(request
+ " not supported (AttributeSummarizer)");
}
}
@Override
protected void renderOffscreenImage(DataSetEvent e) {
if (m_env == null) {
m_env = Environment.getSystemWide();
}
if (m_imageListeners.size() > 0 && !m_processingHeadlessEvents) {
// configure the renderer (if necessary)
setupOffscreenRenderer();
m_offscreenPlotData = new ArrayList();
Instances predictedI = e.getDataSet();
if (predictedI.classIndex() >= 0
&& predictedI.classAttribute().isNominal()) {
// set up multiple series - one for each class
Instances[] classes = new Instances[predictedI.numClasses()];
for (int i = 0; i < predictedI.numClasses(); i++) {
classes[i] = new Instances(predictedI, 0);
classes[i].setRelationName(predictedI.classAttribute().value(i));
}
for (int i = 0; i < predictedI.numInstances(); i++) {
Instance current = predictedI.instance(i);
classes[(int) current.classValue()].add((Instance) current.copy());
}
for (Instances classe : classes) {
m_offscreenPlotData.add(classe);
}
} else {
m_offscreenPlotData.add(new Instances(predictedI));
}
List options = new ArrayList();
String additional = m_additionalOptions;
if (m_additionalOptions != null && m_additionalOptions.length() > 0) {
try {
additional = m_env.substitute(additional);
} catch (Exception ex) {
}
}
if (additional != null && additional.indexOf("-color") < 0) {
// for WekaOffscreenChartRenderer only
if (additional.length() > 0) {
additional += ",";
}
if (predictedI.classIndex() >= 0) {
additional += "-color=" + predictedI.classAttribute().name();
} else {
additional += "-color=/last";
}
}
String[] optionsParts = additional.split(",");
for (String p : optionsParts) {
options.add(p.trim());
}
// only need the x-axis (used to specify the attribute to plot)
String xAxis = m_xAxis;
try {
xAxis = m_env.substitute(xAxis);
} catch (Exception ex) {
}
String width = m_width;
String height = m_height;
int defWidth = 500;
int defHeight = 400;
try {
width = m_env.substitute(width);
height = m_env.substitute(height);
defWidth = Integer.parseInt(width);
defHeight = Integer.parseInt(height);
} catch (Exception ex) {
}
try {
BufferedImage osi = m_offscreenRenderer.renderHistogram(defWidth,
defHeight, m_offscreenPlotData, xAxis, options);
ImageEvent ie = new ImageEvent(this, osi);
notifyImageListeners(ie);
} catch (Exception e1) {
e1.printStackTrace();
}
}
}
public static void main(String[] args) {
try {
if (args.length != 1) {
System.err.println("Usage: AttributeSummarizer ");
System.exit(1);
}
java.io.Reader r = new java.io.BufferedReader(new java.io.FileReader(
args[0]));
Instances inst = new Instances(r);
final javax.swing.JFrame jf = new javax.swing.JFrame();
jf.getContentPane().setLayout(new java.awt.BorderLayout());
final AttributeSummarizer as = new AttributeSummarizer();
as.setInstances(inst);
jf.getContentPane().add(as, java.awt.BorderLayout.CENTER);
jf.addWindowListener(new java.awt.event.WindowAdapter() {
@Override
public void windowClosing(java.awt.event.WindowEvent e) {
jf.dispose();
System.exit(0);
}
});
jf.setSize(830, 600);
jf.setVisible(true);
} catch (Exception ex) {
ex.printStackTrace();
System.err.println(ex.getMessage());
}
}
}