def _inference(self, Xs, mode=None):
self._data = list(self.input_pipeline._format_for_inference(Xs))
self._closed = False
n = len(self._data)
if not getattr(self, "estimator", None):
self.estimator = self.get_estimator()
self._input_fn = lambda: self.input_pipeline._dataset_without_targets(
self._data_generator, train=None
).batch(self.config.batch_size)
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.
After Change
else:
estimator = self.get_estimator()
input_fn = self.input_pipeline.get_predict_input_fn(Xs)
length = len(Xs) if not callable(Xs) else None
predictions = tqdm.tqdm(
estimator.predict(
