Class BaseUnderSamplingPreProcessor

    • Field Detail

      • tbpttWindowSize

        protected int tbpttWindowSize
    • Constructor Detail

      • BaseUnderSamplingPreProcessor

        public BaseUnderSamplingPreProcessor()
    • 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
      • adjustMasks

        public INDArray adjustMasks​(INDArray label,
                                    INDArray labelMask,
                                    int minorityLabel,
                                    double targetDist)