Searched refs:init_probs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sampling_ops.py | 139 init_probs=None, argument
203 if init_probs is None:
207 init_probs = _estimate_data_distribution(
210 init_probs = ops.convert_to_tensor(init_probs, dtype=dtypes.float32)
213 tensor_list, labels, [init_probs, target_probs] = _verify_input(
214 tensor_list, labels, [init_probs, target_probs])
221 math_ops.not_equal(init_probs, 0),
224 'probability: ', init_probs, target_probs
226 init_probs = control_flow_ops.with_dependencies([assert_op], init_probs)
229 accept_probs = _calculate_acceptance_probabilities(init_probs, target_probs)
[all …]
|
| D | sampling_ops_test.py | 46 init_probs = [.1, .3, .1, .3, .2]
56 init_probs,
64 init_probs,
71 probs, batch_size, init_probs)
76 array_ops.zeros([1, 3]), label, probs, batch_size, init_probs)
85 init_probs,
95 init_probs,
100 sampling_ops.stratified_sample(val, label, 1, batch_size, init_probs)
110 init_probs)
116 val, label, [.1] * 10, batch_size, init_probs=[.2] * 5)
[all …]
|