Home
last modified time | relevance | path

Searched refs:categorical_column (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py43 def embedding_column(categorical_column, argument
74 if not isinstance(categorical_column, _SUPPORTED_CATEGORICAL_COLUMNS):
79 ]), type(categorical_column)))
86 categorical_column.name))
91 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
103 categorical_column=categorical_column,
125 for categorical_column in categorical_columns:
126 if not isinstance(categorical_column, _SUPPORTED_CATEGORICAL_COLUMNS):
131 ]), type(categorical_column)))
150 for categorical_column in categorical_columns:
[all …]
Dfeature_column_test.py47 categorical_column = fc_lib.categorical_column_with_identity(
51 categorical_column, dimension=embedding_dimension)
52 self.assertIs(categorical_column, embedding_column.categorical_column)
63 categorical_column = fc_lib.categorical_column_with_identity(
67 categorical_column,
71 self.assertIs(categorical_column, embedding_column.categorical_column)
121 categorical_column = fc_lib.categorical_column_with_identity(
124 categorical_column,
155 self.assertIs(categorical_column_a, embedding_column_a.categorical_column)
156 self.assertIs(categorical_column_b, embedding_column_b.categorical_column)
[all …]
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py818 def embedding_column(categorical_column, argument
906 categorical_column.name))
912 categorical_column=categorical_column,
1056 c0 = c0.categorical_column
1060 c = c.categorical_column
1082 categorical_column=column,
1227 c0 = c0.categorical_column
1230 c = c.categorical_column
1256 categorical_column=column, combiner=combiner, max_norm=max_norm))
1879 def indicator_column(categorical_column): argument
[all …]
Dfeature_column.py810 def _embedding_column(categorical_column, argument
898 categorical_column.name))
903 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
915 categorical_column=categorical_column,
1441 def _indicator_column(categorical_column): argument
1472 return _IndicatorColumn(categorical_column)
1475 def _weighted_categorical_column(categorical_column, argument
1545 categorical_column=categorical_column,
2404 self._name = '{}_embedding'.format(self.categorical_column.name)
2409 return self.categorical_column._parse_example_spec # pylint: disable=protected-access
[all …]
Dfeature_column_v2_test.py3192 categorical_column = fc.categorical_column_with_identity(
3207 categorical_column,
3237 categorical_column = fc.categorical_column_with_identity(
3252 categorical_column,
3816 categorical_column = fc.categorical_column_with_identity(
3831 categorical_column,
3861 categorical_column = fc.categorical_column_with_identity(
3876 categorical_column,
5876 self.assertEqual(indicator_a.categorical_column.name, 'a')
5883 self.assertEqual(indicator_b.categorical_column.name, 'b')
[all …]
Dsequence_feature_column_test.py526 categorical_column = sfc.sequence_categorical_column_with_identity(
528 indicator_column = fc.indicator_column(categorical_column)
972 categorical_column = sfc.sequence_categorical_column_with_identity(
975 categorical_column, dimension=embedding_dimension,
1010 categorical_column = sfc.sequence_categorical_column_with_identity(
1013 categorical_column, dimension=2)
1037 categorical_column = sfc.sequence_categorical_column_with_identity(
1040 categorical_column, dimension=2)
1261 categorical_column = sfc.sequence_categorical_column_with_identity(
1263 indicator_column = fc.indicator_column(categorical_column)
[all …]
Dfeature_column_test.py2651 categorical_column = fc._categorical_column_with_identity(
2665 categorical_column,
2695 categorical_column = fc._categorical_column_with_identity(
2710 categorical_column,
4572 self.assertEqual(indicator_a.categorical_column.name, 'a')
4579 self.assertEqual(indicator_b.categorical_column.name, 'b')
4639 self.assertEqual(column_copy.categorical_column.name, 'a')
4785 categorical_column = fc._categorical_column_with_identity(
4789 categorical_column, dimension=embedding_dimension)
4790 self.assertIs(categorical_column, embedding_column.categorical_column)
[all …]
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column_test.py533 categorical_column = sfc.sequence_categorical_column_with_identity(
535 indicator_column = fc._indicator_column(categorical_column)
920 categorical_column = sfc.sequence_categorical_column_with_identity(
923 categorical_column,
959 categorical_column = sfc.sequence_categorical_column_with_identity(
961 embedding_column = fc._embedding_column(categorical_column, dimension=2)
986 categorical_column = sfc.sequence_categorical_column_with_identity(
988 embedding_column = fc._embedding_column(categorical_column, dimension=2)
1219 categorical_column = sfc.sequence_categorical_column_with_identity(
1221 indicator_column = fc._indicator_column(categorical_column)
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.feature_column.pbtxt29 …argspec: "args=[\'categorical_column\', \'dimension\', \'combiner\', \'initializer\', \'ckpt_to_lo…
33 argspec: "args=[\'categorical_column\'], varargs=None, keywords=None, defaults=None"
69 …argspec: "args=[\'categorical_column\', \'weight_feature_key\', \'dtype\'], varargs=None, keywords…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.feature_column.pbtxt29 …argspec: "args=[\'categorical_column\', \'dimension\', \'combiner\', \'initializer\', \'ckpt_to_lo…
33 argspec: "args=[\'categorical_column\'], varargs=None, keywords=None, defaults=None"
77 …argspec: "args=[\'categorical_column\', \'weight_feature_key\', \'dtype\'], varargs=None, keywords…
/external/tensorflow/tensorflow/contrib/timeseries/examples/
Dlstm.py192 categorical_column = tf.feature_column.categorical_column_with_hash_bucket(
201 categorical_column=categorical_column, dimension=10)]
Dknown_anomaly.py94 categorical_column=string_feature)
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dhead_test.py377 categorical_column = feature_column.categorical_column_with_hash_bucket(
383 categorical_column=categorical_column, dimension=10)]
/external/tensorflow/tensorflow/python/training/
Dwarm_starting_util_test.py1066 categorical_column=sc_vocab,
1136 categorical_column=sc_vocab,
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_test.py622 categorical_column=core_feature_column