org.apache.ignite.ml.preprocessing.encoding

Class EncoderPartitionData

java.lang.Object

org.apache.ignite.ml.preprocessing.encoding.EncoderPartitionData

All Implemented Interfaces:

AutoCloseable

public class EncoderPartitionData
extends Object
implements AutoCloseable

Partition data used in Encoder preprocessor.

Constructor Summary

Constructors

Constructor and Description
EncoderPartitionData() Constructs a new instance of String Encoder partition data.

Method Summary

Modifier and Type	Method and Description
Map<String,Integer>[]	categoryFrequencies() Gets the array of maps of frequencies by value in partition for each feature in the dataset.
void	close()
EncoderPartitionData	withCategoryFrequencies(Map<String,Integer>[] categoryFrequencies) Sets the array of maps of frequencies by value in partition for each feature in the dataset.

Methods inherited from class java.lang.Object

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

Constructor Detail

EncoderPartitionData

public EncoderPartitionData()

Constructs a new instance of String Encoder partition data.

Method Detail

categoryFrequencies

public Map<String,Integer>[] categoryFrequencies()

Gets the array of maps of frequencies by value in partition for each feature in the dataset.

Returns:

The frequencies.

withCategoryFrequencies

public EncoderPartitionData withCategoryFrequencies(Map<String,Integer>[] categoryFrequencies)

Sets the array of maps of frequencies by value in partition for each feature in the dataset.

Parameters:

categoryFrequencies - The given value.

Returns:

The partition data.

close

public void close()

Specified by:

close in interface AutoCloseable