Searched refs:num_label_columns (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | composable_model.py | 57 num_label_columns, argument
78 self._num_label_columns = num_label_columns
160 num_label_columns, argument
188 num_label_columns=num_label_columns,
275 num_label_columns, argument
308 num_label_columns=num_label_columns,
|
| D | rnn_common.py | 200 num_label_columns = activations_shape[2]
204 [-1, num_label_columns])
288 if target_column.num_label_columns == 1:
|
| D | 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,
|
| D | state_saving_rnn_estimator.py | 47 num_label_columns, argument
83 num_outputs=num_label_columns,
497 num_label_columns=target_column.num_label_columns,
|
| D | dynamic_rnn_estimator.py | 216 num_label_columns, argument
254 num_outputs=num_label_columns,
496 target_column.num_label_columns,
|
| D | dynamic_rnn_estimator_test.py | 70 def __init__(self, num_label_columns=None): argument
71 self._num_label_columns = num_label_columns
85 def num_label_columns(self): member in MockTargetColumn
112 num_label_columns=self.NUM_LABEL_COLUMNS)
176 self.mock_target_column.num_label_columns)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | target_column.py | 151 def __init__(self, loss_fn, num_label_columns, label_name, weight_column_name, argument
155 if num_label_columns is None: # n_classes can be 0
159 self._num_label_columns = num_label_columns
181 def num_label_columns(self): member in _TargetColumn
269 num_label_columns=label_dimension,
275 if self.num_label_columns == 1:
299 num_label_columns=1 if n_classes == 2 else n_classes,
305 if self.num_label_columns == 1:
|