2028dffb2a29570c72f10bcb3e35e33fa7fb1c03,finetune/base.py,BaseModel,_inference,#BaseModel#Any#Any#,214

Before Change

    def _inference(self, Xs, mode=None):
        estimator = self.get_estimator()
        input_func = self.input_pipeline.get_predict_input_fn(Xs)
        length = len(Xs) if not callable(Xs) else None

        pred_gen = list(
            map(
                lambda y: y[mode] if mode else y,

After Change

            self._predictions = self.estimator.predict(input_fn=self._input_fn)

        predictions = [None] * n
        for i in range(n):
            y = next(self._predictions)
            y = y[mode] if mode else y
            predictions[i] = y
        return predictions
        
    def fit(self, *args, **kwargs):
         An alias for finetune. 

Instances