72a8442d538cdbffff25ce639aebf10a79db612e,pliers/extractors/models.py,TFHubExtractor,_extract,#TFHubExtractor#Any#,48

Before Change

        
    def _extract(self, stim):
        features = listify(self.get_feature_names())
        input = listify(stim.data)
        if self.transform_fn:
            input = self.transform_fn(input)
        output = self.model.signatures["serving_default"](input).numpy() // or self.model(input).numpy()
        return ExtractorResult(output, stim, self, 
                               features=features)

After Change

        
    def _extract(self, stim):
        features = listify(self._get_feature_names())
        inp = self._preprocess(stim)
        out = self.model(inp).numpy()
        out = self._postprocess(out)
        return ExtractorResult(out, stim, self, 
                               features=features)

Instances