Examples with CompactSampler com.amazon.randomcutforest.sampler.CompactSampler used on opensource projects

Example 6 with CompactSampler

use of com.amazon.randomcutforest.sampler.CompactSampler in project random-cut-forest-by-aws by aws.

the class CompactSamplerMapperTest method nonemptySamplerProvider.

public static Stream<Arguments> nonemptySamplerProvider() {
    CompactSampler fullSampler1 = CompactSampler.builder().capacity(sampleSize).timeDecay(lambda).randomSeed(seed).storeSequenceIndexesEnabled(false).build();
    CompactSampler fullSampler2 = CompactSampler.builder().capacity(sampleSize).timeDecay(lambda).randomSeed(seed).storeSequenceIndexesEnabled(true).build();
    Random random = new Random();
    long baseIndex = 10_000;
    for (int i = 0; i < 100; i++) {
        int pointReference = random.nextInt();
        fullSampler1.update(pointReference, baseIndex + i);
        fullSampler2.update(pointReference, baseIndex + i);
    }
    CompactSampler partiallyFullSampler1 = CompactSampler.builder().capacity(sampleSize).timeDecay(lambda).randomSeed(seed).storeSequenceIndexesEnabled(false).build();
    CompactSampler partiallyFullSampler2 = CompactSampler.builder().capacity(sampleSize).timeDecay(lambda).randomSeed(seed).storeSequenceIndexesEnabled(true).build();
    for (int i = 0; i < sampleSize / 2; i++) {
        int pointReference = random.nextInt();
        partiallyFullSampler1.update(pointReference, baseIndex + i);
        partiallyFullSampler2.update(pointReference, baseIndex + i);
    }
    return Stream.of(Arguments.of("full sampler without sequence indexes", fullSampler1), Arguments.of("full sampler with sequence indexes", fullSampler2), Arguments.of("partially full sampler without sequence indexes", partiallyFullSampler1), Arguments.of("partially full sampler with sequence indexes", partiallyFullSampler2));
}

Example 7 with CompactSampler

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