Home
last modified time | relevance | path

Searched refs:col_factors (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dfactorization_ops_test.py59 wals_model.col_factors,
278 col_factors = [x.eval() for x in wals_model.col_factors]
280 self.assertAllClose(col_factors[0], self._col_factors_0, atol=1e-3)
281 self.assertAllClose(col_factors[1], self._col_factors_1, atol=1e-3)
282 self.assertAllClose(col_factors[2], self._col_factors_2, atol=1e-3)
457 col_factors = [x.eval() for x in wals_model.col_factors]
459 self.assertAllClose(col_factors[0], self._col_factors_0, atol=1e-3)
460 self.assertAllClose(col_factors[1], self._col_factors_1, atol=1e-3)
461 self.assertAllClose(col_factors[2], self._col_factors_2, atol=1e-3)
565 col_factors1 = [x.eval() for x in als_model.col_factors]
[all …]
Dfactorization_ops_test_utils.py98 def calculate_loss(input_mat, row_factors, col_factors, regularization=None, argument
129 gathered_col_factors = array_ops.gather(col_factors, col_indices)
139 col_norm = math_ops.reduce_sum(math_ops.square(col_factors))
141 row_factors, col_factors, transpose_b=True)))
Dwals_test.py276 col_factors = self._model.get_col_factors()
277 self.assertAllClose(col_factors[0], self._col_factors_0, atol=1e-3)
278 self.assertAllClose(col_factors[1], self._col_factors_1, atol=1e-3)
279 self.assertAllClose(col_factors[2], self._col_factors_2, atol=1e-3)
Dfactorization_ops.py304 def col_factors(self): member in WALSModel