org.apache.ignite.ml.dataset.impl.local

Class LocalDatasetBuilder<K,V>

java.lang.Object

org.apache.ignite.ml.dataset.impl.local.LocalDatasetBuilder<K,V>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

All Implemented Interfaces:

DatasetBuilder<K,V>

public class LocalDatasetBuilder<K,V>
extends Object
implements DatasetBuilder<K,V>

A dataset builder that makes LocalDataset. Encapsulate logic of building local dataset such as allocation required data structures and initialization of context part of partitions.

Constructor Summary

Constructors

Constructor and Description
LocalDatasetBuilder(Map<K,V> upstreamMap, IgniteBiPredicate<K,V> filter, int partitions) Constructs a new instance of local dataset builder that makes LocalDataset.
LocalDatasetBuilder(Map<K,V> upstreamMap, int partitions) Constructs a new instance of local dataset builder that makes LocalDataset with default predicate that passes all upstream entries to dataset.

Method Summary

Modifier and Type	Method and Description
<C extends Serializable,D extends AutoCloseable> LocalDataset<C,D>	build(PartitionContextBuilder<K,V,C> partCtxBuilder, PartitionDataBuilder<K,V,C,D> partDataBuilder) Constructs a new instance of Dataset that includes allocation required data structures and initialization of context part of partitions.
DatasetBuilder<K,V>	withFilter(IgniteBiPredicate<K,V> filterToAdd) Returns new instance of DatasetBuilder using conjunction of internal filter and filterToAdd.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LocalDatasetBuilder

public LocalDatasetBuilder(Map<K,V> upstreamMap,
                           int partitions)

Constructs a new instance of local dataset builder that makes LocalDataset with default predicate that passes all upstream entries to dataset.

Parameters:

upstreamMap - Map with upstream data.

partitions - Number of partitions.

LocalDatasetBuilder

public LocalDatasetBuilder(Map<K,V> upstreamMap,
                           IgniteBiPredicate<K,V> filter,
                           int partitions)

Constructs a new instance of local dataset builder that makes LocalDataset.

Parameters:

upstreamMap - Map with upstream data.

filter - Filter for upstream data.

partitions - Number of partitions.

Method Detail

build

public <C extends Serializable,D extends AutoCloseable> LocalDataset<C,D> build(PartitionContextBuilder<K,V,C> partCtxBuilder,
                                                                                PartitionDataBuilder<K,V,C,D> partDataBuilder)

Constructs a new instance of Dataset that includes allocation required data structures and initialization of context part of partitions.

Specified by:

build in interface DatasetBuilder<K,V>

Type Parameters:

C - Type of a partition context.

D - Type of a partition data.

Parameters:

partCtxBuilder - Partition context builder.

partDataBuilder - Partition data builder.

Returns:

Dataset.

withFilter

public DatasetBuilder<K,V> withFilter(IgniteBiPredicate<K,V> filterToAdd)

Returns new instance of DatasetBuilder using conjunction of internal filter and filterToAdd.

Specified by:

withFilter in interface DatasetBuilder<K,V>

Parameters:

filterToAdd - Additional filter.