7f6852042663cb04bab39253a29e48d81c262e28,finetune/base.py,BaseModel,_featurize,#BaseModel#Any#,395

Before Change

        self.mode = PredictMode.PROBAS
        estimator = self.get_estimator()
        input_func = self._get_predict_input_fn(Xs)
        pred_gen = estimator.predict(input_fn=input_func, predict_keys=self.mode)
        return list(pred_gen)

    @abstractmethod
    def featurize(self, *args, **kwargs):

After Change

    def _featurize(self, Xs):
        raw_preds = self._inferrence(Xs, PredictMode.FEATURIZE)
        return np.asarray(raw_preds)

    @abstractmethod
    def featurize(self, *args, **kwargs):
        

Instances