172173174175176177178179180181182
DefaultTreeBuilder treeBuilder = new DefaultTreeBuilder(); treeBuilder.setM(m); Dataset dataset = Dataset.load(getConf(), datasetPath); ForestPredictions callback = isOob ? new ForestPredictions(dataset.nbInstances(), dataset.nblabels()) : null; Builder forestBuilder; if (isPartial) {
199200201202203204205206207208209
@Override protected void setup(Context context) throws IOException, InterruptedException { Configuration conf = context.getConfiguration(); Dataset dataset = Builder.loadDataset(conf); setup(dataset.nblabels()); } /** * Useful when testing */