Class BaseUnderSamplingPreProcessor
- java.lang.Object
-
- org.nd4j.linalg.dataset.api.preprocessor.classimbalance.BaseUnderSamplingPreProcessor
-
- Direct Known Subclasses:
UnderSamplingByMaskingMultiDataSetPreProcessor
,UnderSamplingByMaskingPreProcessor
public abstract class BaseUnderSamplingPreProcessor extends Object
-
-
Field Summary
Fields Modifier and Type Field Description protected int
tbpttWindowSize
-
Constructor Summary
Constructors Constructor Description BaseUnderSamplingPreProcessor()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description INDArray
adjustMasks(INDArray label, INDArray labelMask, int minorityLabel, double targetDist)
void
donotMaskAllMajorityWindows()
If a tbptt segment is all majority class labels default behaviour is to mask all time steps in the segment.void
donotMaskMinorityWindows()
If set will not mask timesteps if they fall in a tbptt segment with at least one minority class label
-
-
-
Method Detail
-
donotMaskAllMajorityWindows
public void donotMaskAllMajorityWindows()
If a tbptt segment is all majority class labels default behaviour is to mask all time steps in the segment. donotMaskAllMajorityWindows() will override the default and mask (1-targetDist)% of the time steps
-
donotMaskMinorityWindows
public void donotMaskMinorityWindows()
If set will not mask timesteps if they fall in a tbptt segment with at least one minority class label
-
-