Method getImportanceStats

JSAT/src/jsat/classifiers/trees/MDI.java:66–83

(TreeLearner model, DataSet<Type> data)

    

    @Override
    public <Type extends DataSet> double[] getImportanceStats(TreeLearner model, DataSet<Type> data)
    {
        double[] features = new double[data.getNumFeatures()];
        
        if(!(data instanceof ClassificationDataSet))
            throw new RuntimeException("MDI currently only supports classification datasets");
        
        List<DataPointPair<Integer>> allData = ((ClassificationDataSet)data).getAsDPPList();
        final int K = ((ClassificationDataSet)data).getClassSize();
        ImpurityScore score = new ImpurityScore(K, im);
        for(DataPointPair<Integer> d : allData)
            score.addPoint(d.getDataPoint(), d.getPair());
        
        visit(model.getTreeNodeVisitor(), score, allData, features, score.getSumOfWeights(), K);
        
        return features;
    }
    
    private void visit(TreeNodeVisitor node, ImpurityScore score, List<DataPointPair<Integer>> data, final double[] features , final double N, final int K)
    {