Searched refs:col_steps (Results 1 – 1 of 1) sorted by relevance
191 def col_steps(self): member in WALSMatrixFactorizationTest275 self._model.fit(input_fn=input_fn, steps=self.col_steps)312 self._model.fit(input_fn=input_fn, steps=self.col_steps)348 self._model.fit(input_fn=input_fn, steps=self.col_steps)375 def col_steps(self): member in WALSMatrixFactorizationTestSweeps