Examples with SequentialForestTraversalExecutor com.amazon.randomcutforest.executor.SequentialForestTraversalExecutor used on opensource projects

Example 1 with SequentialForestTraversalExecutor

use of com.amazon.randomcutforest.executor.SequentialForestTraversalExecutor in project random-cut-forest-by-aws by aws.

the class RandomCutForest method initExecutors.

protected <PointReference, Point> void initExecutors(IStateCoordinator<PointReference, Point> updateCoordinator, ComponentList<PointReference, Point> components) {
    if (parallelExecutionEnabled) {
        traversalExecutor = new ParallelForestTraversalExecutor(components, threadPoolSize);
        updateExecutor = new ParallelForestUpdateExecutor<>(updateCoordinator, components, threadPoolSize);
    } else {
        traversalExecutor = new SequentialForestTraversalExecutor(components);
        updateExecutor = new SequentialForestUpdateExecutor<>(updateCoordinator, components);
    }
}

Example 2 with SequentialForestTraversalExecutor

use of com.amazon.randomcutforest.executor.SequentialForestTraversalExecutor in project random-cut-forest-by-aws by aws.

the class RandomCutForestTest method setUp.

@BeforeEach
public void setUp() {
    dimensions = 2;
    sampleSize = 256;
    numberOfTrees = 10;
    components = new ComponentList<>();
    for (int i = 0; i < numberOfTrees; i++) {
        CompactSampler sampler = mock(CompactSampler.class);
        when(sampler.getCapacity()).thenReturn(sampleSize);
        RandomCutTree tree = mock(RandomCutTree.class);
        components.add(spy(new SamplerPlusTree<>(sampler, tree)));
    }
    updateCoordinator = spy(new PointStoreCoordinator<>(new PointStore.Builder().dimensions(2).capacity(1).build()));
    traversalExecutor = spy(new SequentialForestTraversalExecutor(components));
    updateExecutor = spy(new SequentialForestUpdateExecutor<>(updateCoordinator, components));
    RandomCutForest.Builder<?> builder = RandomCutForest.builder().dimensions(dimensions).numberOfTrees(numberOfTrees).sampleSize(sampleSize);
    forest = spy(new RandomCutForest(builder, updateCoordinator, components, builder.getRandom()));
    Whitebox.setInternalState(forest, "traversalExecutor", traversalExecutor);
    Whitebox.setInternalState(forest, "updateExecutor", updateExecutor);
}