Lines Matching refs:input_fn
82 def input_fn(self, np_matrix, batch_size, mode,
85 """Returns an input_fn that selects row and col batches from np_matrix.
258 input_fn = self.input_fn(np_matrix=self.input_matrix,
262 self._model.fit(input_fn=input_fn, steps=self.row_steps)
269 input_fn = self.input_fn(np_matrix=self.input_matrix,
273 self._model.fit(input_fn=input_fn, steps=self.col_steps)
280 input_fn = self.input_fn(np_matrix=self.input_matrix,
286 proj_input_fn = self.input_fn(
293 self._model.fit(input_fn=input_fn, steps=self.row_steps)
303 proj_input_fn = self.input_fn(
310 self._model.fit(input_fn=input_fn, steps=self.col_steps)
323 input_fn = self.input_fn(np_matrix=self.input_matrix,
328 self._model.fit(input_fn=input_fn, steps=self.row_steps)
329 eval_input_fn_row = self.input_fn(np_matrix=self.input_matrix,
335 input_fn=eval_input_fn_row, steps=self._num_rows)['loss']
346 self._model.fit(input_fn=input_fn, steps=self.col_steps)
347 eval_input_fn_col = self.input_fn(np_matrix=self.input_matrix,
353 input_fn=eval_input_fn_col, steps=self._num_cols)['loss']
