HomeSort by relevance Sort by last modified time
    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