223b817345f40076b2ff8dd0e7e49c00e1d6c03c,nussl/separation/primitive/repet_sim.py,RepetSim,__init__,#RepetSim#,40

Before Change

                 max_repeating_frames=None, high_pass_cutoff=None, do_mono=False,
                 use_librosa_stft=constants.USE_LIBROSA_STFT, matlab_fidelity=False,
                 mask_type=mask_separation_base.MaskSeparationBase.SOFT_MASK, mask_threshold=0.5):
        super(RepetSim, self).__init__(input_audio_signal=input_audio_signal, mask_type=mask_type,
                                       mask_threshold=mask_threshold)

        self.high_pass_cutoff = 100 if high_pass_cutoff is None else high_pass_cutoff
        self.similarity_threshold = 0 if similarity_threshold is None else similarity_threshold
        self.min_distance_between_frames = 1 if min_distance_between_frames is None else min_distance_between_frames
        self.max_repeating_frames = 100 if max_repeating_frames is None else max_repeating_frames

        self._min_distance_converted_to_hops = False

        self.verbose = False
        self.similarity_matrix = None
        self.background = None
        self.foreground = None
        self.similarity_indices = None
        self.magnitude_spectrogram = None
        self.stft = None
        self.result_masks = None
        self.use_librosa_stft = use_librosa_stft
        self.matlab_fidelity = matlab_fidelity

        if self.matlab_fidelity:
            self.use_librosa_stft = False

        self.do_mono = do_mono

        if self.do_mono:
            self.audio_signal.to_mono(overwrite=True)

    def run(self):
        

After Change

    def __init__(self, input_audio_signal, similarity_threshold=0, 
                 min_distance_between_frames=1, max_repeating_frames=100, 
                 high_pass_cutoff=100, mask_type="soft", mask_threshold=0.5):
        super().__init__(
            input_audio_signal=input_audio_signal, 
            mask_type=mask_type,
            mask_threshold=mask_threshold)

        self.high_pass_cutoff = high_pass_cutoff
        self.similarity_threshold = similarity_threshold
        self.min_distance_between_frames = min_distance_between_frames

Instances