OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:weighted_ids
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/contrib/layers/python/layers/
feature_column_test.py
102
weighted_ids
= fc.weighted_sparse_column(ids, "weights")
103
self.assertEqual(
weighted_ids
.name, "ids_weighted_by_weights")
108
weighted_ids
= fc.weighted_sparse_column(ids, "weights")
109
self.assertEqual(
weighted_ids
.name, "ids_weighted_by_weights")
110
self.assertEqual(
weighted_ids
.lookup_config, ids.lookup_config)
111
self.assertEqual(
weighted_ids
.lookup_config.vocab_size, 3)
112
self.assertEqual(
weighted_ids
.lookup_config.num_oov_buckets, 7)
113
self.assertEqual(
weighted_ids
.lookup_config.vocabulary_file, "a_file")
327
weighted_ids
= fc.weighted_sparse_column(ids, "weights")
328
one_hot = fc.one_hot_column(
weighted_ids
)
[
all
...]
feature_column_ops_test.py
301
weighted_ids
= feature_column.weighted_sparse_column(ids, "weights")
309
features=features, feature_columns=[
weighted_ids
])
311
self.assertIn(
weighted_ids
, output)
315
self.assertAllEqual(output[
weighted_ids
][0].dense_shape.eval(),
317
self.assertAllEqual(output[
weighted_ids
][0].indices.eval(),
319
self.assertAllEqual(output[
weighted_ids
][0].values.eval(), [2, 2, 0])
320
self.assertAllEqual(output[
weighted_ids
][1].dense_shape.eval(),
322
self.assertAllEqual(output[
weighted_ids
][1].indices.eval(),
324
self.assertEqual(output[
weighted_ids
][1].values.dtype, dtypes.float32)
325
self.assertAllEqual(output[
weighted_ids
][1].values.eval()
[
all
...]
Completed in 88 milliseconds