Search in sources :

Example 1 with DefaultParallelismStrategy

use of org.apache.ignite.ml.environment.parallelism.DefaultParallelismStrategy in project ignite by apache.

the class LearningEnvironmentBuilderTest method withParallelismStrategy.

/**
 */
@Test
public void withParallelismStrategy() {
    assertTrue(LearningEnvironmentBuilder.defaultBuilder().withParallelismStrategyDependency(part -> NoParallelismStrategy.INSTANCE).buildForTrainer().parallelismStrategy() instanceof NoParallelismStrategy);
    assertTrue(LearningEnvironmentBuilder.defaultBuilder().withParallelismStrategyDependency(part -> new DefaultParallelismStrategy()).buildForTrainer().parallelismStrategy() instanceof DefaultParallelismStrategy);
}
Also used : DefaultParallelismStrategy(org.apache.ignite.ml.environment.parallelism.DefaultParallelismStrategy) NoParallelismStrategy(org.apache.ignite.ml.environment.parallelism.NoParallelismStrategy) Test(org.junit.Test)

Aggregations

DefaultParallelismStrategy (org.apache.ignite.ml.environment.parallelism.DefaultParallelismStrategy)1 NoParallelismStrategy (org.apache.ignite.ml.environment.parallelism.NoParallelismStrategy)1 Test (org.junit.Test)1