tclass.WekaBridge.java Source code

Introduction

Here is the source code for tclass.WekaBridge.java

Source

/*
 *   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 <http://www.gnu.org/licenses/>.
 */

package tclass;

import weka.core.Attribute;
import weka.core.DenseInstance;
import weka.core.FastVector;
import weka.core.Instance;
import weka.core.Instances;

/**
  * Class for converting from our interna formats to Weka's. 
  * I'd rather keep things as separate as possible. 
  * 
  * @author Waleed Kadous
  * @version $Id: WekaBridge.java,v 1.1.1.1 2002/06/28 07:36:16 waleed Exp $
  */

public class WekaBridge {

    /** 
     *  Makes an instances objects. 
     */

    public static Instances makeInstances(ClassStreamEventsVecI csevi, ClassDescVecI cdvi, EventDescVecI edvi,
            int evPos, boolean ignoreClasses, boolean ignoreTime) throws Exception {
        // System.out.println("Asked to ignore classes."); 

        Instances retval;
        // Ok, first convert attributes. 
        StreamEventsVecI sevi = csevi.getStreamEventsVec();
        ClassificationVecI cvi = csevi.getClassVec();
        // System.out.println("Event Desc Vec = " + edvi); 
        int numAtts = edvi.elAt(evPos).numParams();
        if (ignoreTime) {
            numAtts = numAtts - 1;
        }
        FastVector atts = makeAttVector(edvi.elAt(evPos), ignoreTime);
        if (!ignoreClasses) {
            int size = cdvi.size();
            FastVector classes = new FastVector(size);
            for (int i = 0; i < size; i++) {
                classes.addElement(cdvi.getClassLabel(i));
            }
            atts.addElement(new Attribute("class", classes));
        }
        int size = csevi.size();
        retval = new Instances(edvi.elName(evPos), atts, size);
        if (!ignoreClasses) {
            retval.setClassIndex(numAtts);
        } else {
            retval.setClassIndex(-1);
        }
        for (int i = 0; i < size; i++) {
            // Get events of this type: 
            EventVecI evi = sevi.elAt(i).getEvents(evPos);
            int numEvents = evi.size();
            for (int j = 0; j < numEvents; j++) {
                // System.out.println("Adding event " + j + " of stream " + i); 
                Instance thisInst = new DenseInstance(atts.size());
                thisInst.setDataset(retval);
                EventI thisEvent = evi.elAt(j);

                for (int k = (ignoreTime ? 1 : 0); k < edvi.elAt(evPos).numParams(); k++) {
                    thisInst.setValue(k - (ignoreTime ? 1 : 0), thisEvent.valOf(k));
                }
                if (!ignoreClasses) {
                    thisInst.setValue(numAtts, cdvi.getClassLabel(cvi.elAt(i).getRealClass()));
                }
                retval.add(thisInst);
            }
        }
        return retval;
    }

    public static Instance makeInstance(EventI ei, EventDescI edi, boolean ignoreTime) {
        // A speed hack. My code is getting pretty f*ing ugly. 
        int numParams = edi.numParams();
        double[] eventdbl = new double[numParams];
        for (int i = (ignoreTime ? 1 : 0); i < numParams; i++) {
            eventdbl[i - (ignoreTime ? 1 : 0)] = (double) ei.valOf(i);
        }
        Instance retval = new DenseInstance(1.0, eventdbl);
        return retval;
    }

    public static Instances makeInstances(ClassStreamAttValVecI csavvi, String name) throws Exception {
        StreamAttValVecI origData = csavvi.getStreamAttValVec();
        AttDescVecI format = origData.getDescription();
        ClassificationVecI classes = csavvi.getClassVec();
        ClassDescVecI classInfo = classes.getClassDescVec();
        FastVector instanceDesc = makeAttVector(format, classInfo);
        int numInstances = origData.size();
        int numAtts = format.size();
        Instances retval = new Instances(name, instanceDesc, numInstances);
        retval.setClassIndex(numAtts); // Set class to last attribute. 

        for (int i = 0; i < numInstances; i++) {
            Instance thisInst = new DenseInstance(numAtts + 1); // To include the class.  
            thisInst.setDataset(retval);
            StreamAttValI thisStream = origData.elAt(i);
            for (int j = 0; j < numAtts; j++) {
                thisInst.setValue(j, thisStream.getAtt(j));
            }
            thisInst.setValue(numAtts, classInfo.getClassLabel(classes.elAt(i).getRealClass()));
            retval.add(thisInst);
        }
        return retval;
    }

    public static FastVector makeAttVector(EventDescI edi, boolean ignoreTime) {
        int size = edi.numParams();
        if (ignoreTime) {
            size--;
        }
        FastVector retval = new FastVector(size);
        int i;
        for (i = (ignoreTime ? 1 : 0); i < edi.numParams(); i++) {
            retval.addElement(makeAtt(edi.paramName(i), edi.getDataType(i)));
        }
        return retval;
    }

    public static FastVector makeAttVector(AttDescVecI advi, ClassDescVecI classInfo) {
        int numAtts = advi.size();
        FastVector retval = new FastVector(numAtts + 1);
        for (int i = 0; i < numAtts; i++) {
            retval.addElement(makeAtt(advi.elAt(i)));
        }
        int numClasses = classInfo.size();
        FastVector classes = new FastVector(numClasses);
        for (int i = 0; i < numClasses; i++) {
            classes.addElement(classInfo.getClassLabel(i));
        }
        retval.addElement(new Attribute("class", classes));
        return retval;
    }

    public static Attribute makeAtt(AttDescI adi) {
        return makeAtt(adi.getName(), adi.getDataType());
    }

    /** Hacked version!! Only handles numeric attributes in events,
     * which is dodgy. 
     */
    public static Attribute makeAtt(String name, DataTypeI dt) {
        return new Attribute(name);
    }
}