c87070113341a10fe35e4a3d6ff25c8e54d69feb,finetune/base.py,BaseModel,_predict,#BaseModel#Any#Any#,420

Before Change

    def _predict(self, Xs, context=None, **kwargs):
        raw_preds = self._inference(Xs, predict_keys=[PredictMode.NORMAL], context=context, **kwargs)
        return self.input_pipeline.label_encoder.inverse_transform(
            np.asarray(raw_preds)
        )

    def predict(self, Xs, context=None, **kwargs):
        if self.config.sort_by_length:
            Xs, invert_idxs = self._sort_by_length(Xs)

After Change

        return self.finetune(*args, **kwargs)

    def _predict(self, Xs, context=None, **kwargs):
        raise NotImplemented()

    def predict(self, Xs, context=None, **kwargs):
        if self.config.sort_by_length:
            Xs, invert_idxs = self._sort_by_length(Xs)

Instances