1ddc26b3cfc16daae7cfdde9a730be33b8b690c0,pliers/extractors/text.py,PretrainedBertExtractor,_extract,#PretrainedBertExtractor#Any#,484

Before Change

    def _extract(self, stims):
        t_wds, t_ons, t_dur, t_tok, t_idx = self._preprocess(stims)
        preds = self.model(t_idx)
        preds = [p.detach().numpy() if self.framework == "pt" else
                 p.numpy() for p in preds]
        out, t_tok, t_wds, t_ons, t_dur = self._postprocess(preds, t_tok, t_wds, t_ons, t_dur)
        data = [out.tolist(), t_tok, t_wds]
        features = self._get_feature_names()
        return ExtractorResult(data, stims, self, features=features,
                               onsets=t_ons, durations=t_dur)

After Change

        wds, ons, dur, tok, idx = self._preprocess(stims)
        preds = self.model(idx)
        preds = [p.detach() if self.framework == "pt" else p for p in preds]
        preds = [p.numpy().squeeze() for p in preds]
        data, ons, dur = self._postprocess(preds, tok, wds, ons, dur)
        features = self._get_feature_names()
        return ExtractorResult(data, stims, self, 
                               features=features, onsets=ons, durations=dur)

Instances