Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py55 def embedding_column(categorical_column, argument
109 if isinstance(categorical_column, _DENYLISTED_CATEGORICAL_COLUMNS_V2):
112 type(categorical_column))
113 if not isinstance(categorical_column, _SUPPORTED_CATEGORICAL_COLUMNS):
118 ]), type(categorical_column)))
125 categorical_column.name))
130 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
142 categorical_column=categorical_column,
225 for categorical_column in categorical_columns:
226 if isinstance(categorical_column, _DENYLISTED_CATEGORICAL_COLUMNS_V2):
[all …]
Dfeature_column_v2.py55 def embedding_column_v2(categorical_column, argument
148 if not isinstance(categorical_column, _SUPPORTED_CATEGORICAL_COLUMNS_V2):
153 ]), type(categorical_column)))
163 categorical_column.name))
184 if isinstance(categorical_column, _SUPPORTED_SEQUENCE_COLUMNS):
190 categorical_column=categorical_column,
199 categorical_column=categorical_column,
317 for categorical_column in categorical_columns:
318 if not isinstance(categorical_column, _SUPPORTED_CATEGORICAL_COLUMNS_V2):
323 ]), type(categorical_column)))
[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)
64 categorical_column = fc_lib.categorical_column_with_hash_bucket(
68 tpu_fc.embedding_column(categorical_column, dimension=embedding_dimension)
73 categorical_column = fc_lib.categorical_column_with_identity(
77 categorical_column, dimension=embedding_dimension)
78 self.assertIs(categorical_column, embedding_column.categorical_column)
88 categorical_column = fc_lib.categorical_column_with_identity(
92 categorical_column,
[all …]
Dfeature_column_v2_test.py86 categorical_column = fc_lib.categorical_column_with_identity(
90 categorical_column, dimension=embedding_dimension)
92 self.assertIs(categorical_column, embedding_column.categorical_column)
99 categorical_column = fc_lib.categorical_column_with_identity(
103 categorical_column,
107 self.assertIs(categorical_column, embedding_column.categorical_column)
175 categorical_column = fc_lib.categorical_column_with_identity(
181 categorical_column,
222 categorical_column = fc_lib.categorical_column_with_identity(
225 categorical_column, dimension=2)
[all …]
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py491 def embedding_column(categorical_column, argument
587 categorical_column.name))
593 categorical_column=categorical_column,
747 c0 = c0.categorical_column
752 c = c.categorical_column
774 categorical_column=column,
928 c0 = c0.categorical_column
931 c = c.categorical_column
958 categorical_column=column, combiner=combiner, max_norm=max_norm))
1616 def indicator_column(categorical_column): argument
[all …]
Dfeature_column.py822 def _embedding_column(categorical_column, argument
918 categorical_column.name))
923 embedding_shape = categorical_column._num_buckets, dimension # pylint: disable=protected-access
935 categorical_column=categorical_column,
1462 def _indicator_column(categorical_column): argument
1493 return _IndicatorColumn(categorical_column)
1496 def _weighted_categorical_column(categorical_column, argument
1566 categorical_column=categorical_column,
2486 categorical_column, argument
2497 categorical_column=categorical_column,
[all …]
Dsequence_feature_column_test.py209 self.assertIsNot(parent, new_animal.categorical_column)
217 self.assertIs(parent, new_animal.categorical_column)
440 categorical_column = sfc.sequence_categorical_column_with_identity(
443 categorical_column, dimension=embedding_dimension,
478 categorical_column = sfc.sequence_categorical_column_with_identity(
481 categorical_column, dimension=2)
505 categorical_column = sfc.sequence_categorical_column_with_identity(
508 categorical_column, dimension=2)
729 categorical_column = sfc.sequence_categorical_column_with_identity(
731 indicator_column = fc.indicator_column(categorical_column)
[all …]
Dfeature_column_v2_test.py2309 categorical_column = fc.categorical_column_with_identity(
2324 categorical_column,
2354 categorical_column = fc.categorical_column_with_identity(
2369 categorical_column,
4081 categorical_column = fc.categorical_column_with_identity(
4084 categorical_column,
4124 categorical_column = fc.categorical_column_with_identity(
4127 categorical_column,
4323 self.assertEqual(indicator_a.categorical_column.name, 'a')
4330 self.assertEqual(indicator_b.categorical_column.name, 'b')
[all …]
Dfeature_column_test.py2623 categorical_column = fc._categorical_column_with_identity(
2637 categorical_column,
2667 categorical_column = fc._categorical_column_with_identity(
2682 categorical_column,
4362 categorical_column = fc._categorical_column_with_identity(
4365 categorical_column,
4399 categorical_column = fc._categorical_column_with_identity(
4402 categorical_column,
4568 self.assertEqual(indicator_a.categorical_column.name, 'a')
4575 self.assertEqual(indicator_b.categorical_column.name, 'b')
[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…
Dtensorflow.tpu.experimental.pbtxt37 …argspec: "args=[\'categorical_column\', \'dimension\', \'combiner\', \'initializer\', \'max_sequen…
/external/tensorflow/tensorflow/python/keras/feature_column/
Ddense_features_test.py69 categorical_column = fc.categorical_column_with_identity(
84 categorical_column,
114 categorical_column = fc.categorical_column_with_identity(
133 categorical_column,
166 categorical_column = fc.categorical_column_with_identity(
181 categorical_column,
770 categorical_column = fc.categorical_column_with_identity(
777 categorical_column,
864 categorical_column = fc.categorical_column_with_identity(
867 categorical_column,
[all …]
Ddense_features_v2_test.py64 categorical_column = fc.categorical_column_with_identity(
79 categorical_column,
109 categorical_column = fc.categorical_column_with_identity(
124 categorical_column,
Dsequence_feature_column_test.py527 categorical_column = sfc.sequence_categorical_column_with_identity(
529 indicator_column = fc.indicator_column(categorical_column)
/external/tensorflow/tensorflow/python/training/
Dwarm_starting_util_test.py1068 categorical_column=sc_vocab,
1138 categorical_column=sc_vocab,