Searched refs:cols_to_vars (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/estimator/ |
| D | warm_starting_util_test.py | 83 cols_to_vars = {}
90 cols_to_vars=cols_to_vars)
92 return cols_to_vars
94 def _assert_cols_to_vars(self, cols_to_vars, cols_to_expected_values, sess): argument
96 for i, var in enumerate(cols_to_vars[col]):
322 cols_to_vars = self._create_linear_model([sc_int], partitioner)
326 self._assert_cols_to_vars(cols_to_vars, {sc_int: [np.zeros([10, 1])]},
332 cols_to_vars = self._create_linear_model([sc_int], partitioner)
338 self._assert_cols_to_vars(cols_to_vars, {sc_int: [prev_int_val]}, sess)
353 cols_to_vars = self._create_linear_model([sc_hash], partitioner)
[all …]
|
| /external/tensorflow/tensorflow/python/estimator/canned/ |
| D | linear.py | 50 def _compute_fraction_of_zero(cols_to_vars): argument
61 for var_or_var_list in cols_to_vars.values():
94 cols_to_vars = {}
99 cols_to_vars=cols_to_vars)
100 bias = cols_to_vars.pop('bias')
108 _compute_fraction_of_zero(cols_to_vars))
|
| D | linear_testing_utils.py | 1848 cols_to_vars = {}
1856 cols_to_vars=cols_to_vars)
1857 cols_to_vars.pop('bias')
1858 fraction_zero = linear._compute_fraction_of_zero(cols_to_vars)
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 168 cols_to_vars=None, argument
204 if cols_to_vars is not None:
207 cols_to_vars[column] = ops.get_collection(
219 cols_to_vars=None): argument
274 trainable, cols_to_vars)
288 cols_to_vars=None): argument
294 self._cols_to_vars = cols_to_vars
307 cols_to_vars=None,
342 cols_to_vars=None): argument
435 if cols_to_vars is not None:
[all …]
|
| D | feature_column_test.py | 1380 cols_to_vars = {}
1381 fc.linear_model(features, [price1, price2], cols_to_vars=cols_to_vars)
1385 self.assertAllEqual(cols_to_vars['bias'], [bias])
1386 self.assertAllEqual(cols_to_vars[price1], [price1_var])
1387 self.assertAllEqual(cols_to_vars[price2], [price2_var])
1397 cols_to_vars = {}
1401 fc.linear_model(features, [price1, price2], cols_to_vars=cols_to_vars)
1403 self.assertEqual([0.], cols_to_vars['bias'][0].eval())
1405 self.assertAllEqual([[0.]], cols_to_vars[price1][0].eval())
1406 self.assertAllEqual([[0.]], cols_to_vars[price1][1].eval())
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.feature_column.pbtxt | 37 …ures\', \'feature_columns\', \'weight_collections\', \'trainable\', \'cols_to_vars\'], varargs=Non…
41 …nits\', \'sparse_combiner\', \'weight_collections\', \'trainable\', \'cols_to_vars\'], varargs=Non…
|