org.apache.ignite.ml.dataset.primitive.builder.data

Class SimpleLabeledDatasetDataBuilder<K,V,C extends Serializable>

java.lang.Object

org.apache.ignite.ml.dataset.primitive.builder.data.SimpleLabeledDatasetDataBuilder<K,V,C>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

C - type of a partition context.

All Implemented Interfaces:

Serializable, PartitionDataBuilder<K,V,C,SimpleLabeledDatasetData>

public class SimpleLabeledDatasetDataBuilder<K,V,C extends Serializable>
extends Object
implements PartitionDataBuilder<K,V,C,SimpleLabeledDatasetData>

A partition data builder that makes SimpleLabeledDatasetData.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
SimpleLabeledDatasetDataBuilder(IgniteBiFunction<K,V,Vector> featureExtractor, IgniteBiFunction<K,V,double[]> lbExtractor) Constructs a new instance of partition data builder that makes SimpleLabeledDatasetData.

Method Summary

Modifier and Type	Method and Description
SimpleLabeledDatasetData	build(Iterator<UpstreamEntry<K,V>> upstreamData, long upstreamDataSize, C ctx) Builds a new partition data from a partition upstream data and partition context

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.ignite.ml.dataset.PartitionDataBuilder

andThen

Constructor Detail

SimpleLabeledDatasetDataBuilder

public SimpleLabeledDatasetDataBuilder(IgniteBiFunction<K,V,Vector> featureExtractor,
                                       IgniteBiFunction<K,V,double[]> lbExtractor)

Constructs a new instance of partition data builder that makes SimpleLabeledDatasetData.

Parameters:

featureExtractor - Function that extracts features from an upstream data.

lbExtractor - Function that extracts labels from an upstream data.

Method Detail

build

public SimpleLabeledDatasetData build(Iterator<UpstreamEntry<K,V>> upstreamData,
                                      long upstreamDataSize,
                                      C ctx)

Builds a new partition data from a partition upstream data and partition context

Specified by:

build in interface PartitionDataBuilder<K,V,C extends Serializable,SimpleLabeledDatasetData>

Parameters:

upstreamData - Partition upstream data.

upstreamDataSize - Partition upstream data size.

ctx - Partition context.

Returns:

Partition data.