| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| target_column.py | 139 num_label_columns: Integer, number of label columns.
150 def __init__(self, loss_fn, num_label_columns, label_name, weight_column_name,
154 if num_label_columns is None: # n_classes can be 0
155 raise ValueError("num_label_columns must be provided")
158 self._num_label_columns = num_label_columns
180 def num_label_columns(self): member in class:_TargetColumn
266 num_label_columns=label_dimension,
272 if self.num_label_columns == 1:
296 num_label_columns=1 if n_classes == 2 else n_classes,
302 if self.num_label_columns == 1
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| composable_model.py | 50 num_label_columns,
59 num_label_columns: The number of label columns.
71 self._num_label_columns = num_label_columns
149 num_label_columns,
159 num_label_columns: The number of label columns.
177 num_label_columns=num_label_columns,
260 num_label_columns,
272 num_label_columns: The number of label columns.
293 num_label_columns=num_label_columns
[all...] |
| composable_model_test.py | 80 num_label_columns=head.logits_dimension),
94 num_label_columns=head.logits_dimension, _joint_weights=True),
108 num_label_columns=head.logits_dimension,
|
| rnn_common.py | 195 num_label_columns = activations_shape[2]
199 [-1, num_label_columns])
283 if target_column.num_label_columns == 1:
|
| state_saving_rnn_estimator.py | 42 num_label_columns,
52 num_label_columns: The desired output dimension.
62 activations: The output of the RNN, projected to `num_label_columns`
63 dimensions, a `Tensor` of shape `[batch_size, T, num_label_columns]`.
78 num_outputs=num_label_columns,
492 num_label_columns=target_column.num_label_columns,
|
| dynamic_rnn_estimator.py | 211 num_label_columns,
223 num_label_columns: The desired output dimension.
233 activations: The output of the RNN, projected to `num_label_columns`
249 num_outputs=num_label_columns,
491 target_column.num_label_columns,
|
| dynamic_rnn_estimator_test.py | 70 def __init__(self, num_label_columns=None):
71 self._num_label_columns = num_label_columns
85 def num_label_columns(self): member in class:MockTargetColumn
87 raise ValueError('MockTargetColumn.num_label_columns has not been set.')
104 NUM_LABEL_COLUMNS = 6
106 'inputs', dimension=NUM_LABEL_COLUMNS)
112 num_label_columns=self.NUM_LABEL_COLUMNS)
176 self.mock_target_column.num_label_columns)
184 expected_activations_shape = np.array([3, 2, self.NUM_LABEL_COLUMNS])
[all...] |
