Class AttributeBasedStratifier
- java.lang.Object
-
- ai.libs.jaicore.ml.core.filter.sampling.inmemory.stratified.sampling.AttributeBasedStratifier
-
- All Implemented Interfaces:
IStratifier,org.api4.java.common.control.ILoggingCustomizable,org.api4.java.common.control.IParallelizable
public class AttributeBasedStratifier extends java.lang.Object implements IStratifier, org.api4.java.common.control.ILoggingCustomizable
This class is responsible for computing the amount of strati in attribute-based stratified sampling and assigning elements to the strati.
-
-
Constructor Summary
Constructors Constructor Description AttributeBasedStratifier()AttributeBasedStratifier(java.util.List<java.lang.Integer> attributeIndices)AttributeBasedStratifier(java.util.List<java.lang.Integer> attributeIndices, DiscretizationHelper.DiscretizationStrategy discretizationStrategy, int numberOfCategories)AttributeBasedStratifier(java.util.List<java.lang.Integer> attributeIndices, java.util.Map<java.lang.Integer,AttributeDiscretizationPolicy> discretizationPolicies)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description intcreateStrati(org.api4.java.ai.ml.core.dataset.IDataset<?> dataset)Prepares the stratification technique but does not assign instances to strati.java.lang.StringgetLoggerName()intgetNumCPUs()intgetStratum(org.api4.java.ai.ml.core.dataset.IInstance datapoint)Determines to which stratum this instance belongsvoidsetLoggerName(java.lang.String name)voidsetNumCPUs(int numberOfCPUs)
-
-
-
Constructor Detail
-
AttributeBasedStratifier
public AttributeBasedStratifier()
-
AttributeBasedStratifier
public AttributeBasedStratifier(java.util.List<java.lang.Integer> attributeIndices)
-
AttributeBasedStratifier
public AttributeBasedStratifier(java.util.List<java.lang.Integer> attributeIndices, DiscretizationHelper.DiscretizationStrategy discretizationStrategy, int numberOfCategories)
-
AttributeBasedStratifier
public AttributeBasedStratifier(java.util.List<java.lang.Integer> attributeIndices, java.util.Map<java.lang.Integer,AttributeDiscretizationPolicy> discretizationPolicies)
-
-
Method Detail
-
createStrati
public int createStrati(org.api4.java.ai.ml.core.dataset.IDataset<?> dataset)
Description copied from interface:IStratifierPrepares the stratification technique but does not assign instances to strati.- Specified by:
createStratiin interfaceIStratifier- Returns:
- The number of strati for the given dataset
-
setNumCPUs
public void setNumCPUs(int numberOfCPUs)
- Specified by:
setNumCPUsin interfaceorg.api4.java.common.control.IParallelizable
-
getNumCPUs
public int getNumCPUs()
- Specified by:
getNumCPUsin interfaceorg.api4.java.common.control.IParallelizable
-
getStratum
public int getStratum(org.api4.java.ai.ml.core.dataset.IInstance datapoint)
Description copied from interface:IStratifierDetermines to which stratum this instance belongs- Specified by:
getStratumin interfaceIStratifier- Returns:
- id of stratum
-
getLoggerName
public java.lang.String getLoggerName()
- Specified by:
getLoggerNamein interfaceorg.api4.java.common.control.ILoggingCustomizable
-
setLoggerName
public void setLoggerName(java.lang.String name)
- Specified by:
setLoggerNamein interfaceorg.api4.java.common.control.ILoggingCustomizable
-
-