Searched refs:score_fn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils.cc | 45 void GetTwoBest(int max, const std::function<float(int)>& score_fn, in GetTwoBest() argument
53 float score = score_fn(i); in GetTwoBest()
105 std::function<float(int)> score_fn = in GetTwoBestClassification() local
109 GetTwoBest(num_splits, score_fn, best_score, best_index, second_best_score, in GetTwoBestClassification()
|
| D | tree_utils.h | 59 void GetTwoBest(int max, const std::function<float(int)>& score_fn,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.cc | 305 std::function<float(int)> score_fn = in CheckFinishEarlyHoeffding() local
313 GetTwoBest(num_splits(), score_fn, &best_score, &best_index, in CheckFinishEarlyHoeffding()
343 std::function<float(int)> score_fn = in CheckFinishEarlyBootstrap() local
351 GetTwoBest(num_splits(), score_fn, &best_score, &best_index, in CheckFinishEarlyBootstrap()
|