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

Example 6 with SamplerPlusTree

use of com.amazon.randomcutforest.executor.SamplerPlusTree 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);
}