bdf66f4cc11c083b48e74efa4df874668ef74805,pliers/extractors/text.py,WordCounterExtractor,_extract,#WordCounterExtractor#Any#,423

Before Change

        word_counter = pd.Series(tokens).groupby(tokens).cumcount() + 1
        if self.log_scale:
            word_counter = np.log(word_counter)
        results = []
        for i, count in enumerate(word_counter):
            results.append(ExtractorResult([count], list(stims)[i],
                           self, features=self.features))
        return results
        

After Change

    def _extract(self, stims):
        
        onsets = [s.onset for s in stims]
        durations = [s.duration for s in stims]
        tokens = [s.text for s in stims]
        tokens = [t if self.case_sensitive else t.lower() for t in tokens]
        word_counter = pd.Series(tokens).groupby(tokens).cumcount() + 1
        if self.log_scale:
            word_counter = np.log(word_counter)

        return ExtractorResult([word_counter], stims, self,
                               features=self.features,
                               onsets=onsets, durations=durations)

        

Instances