Searched refs:segment_label_sorter (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/objective/ |
| H A D | test_ranking_obj_gpu.cu | 135 auto segment_label_sorter = RankSegmentSorterTestImpl<float>( in TEST() local
166 std::vector<float> hgroup_dcgs(segment_label_sorter->GetNumGroups()); in TEST()
168 std::vector<uint32_t> hgroups(segment_label_sorter->GetNumGroups() + 1); in TEST()
170 EXPECT_EQ(hgroup_dcgs.size(), segment_label_sorter->GetNumGroups()); in TEST()
171 std::vector<float> hsorted_labels(segment_label_sorter->GetNumItems()); in TEST()
191 auto segment_label_sorter = RankSegmentSorterTestImpl<float>( in TEST() local
201 segment_label_sorter->CreateIndexableSortedPositions(); in TEST()
202 std::vector<uint32_t> sorted_indices(segment_label_sorter->GetNumItems()); in TEST()
218 auto segment_label_sorter = RankSegmentSorterTestImpl<float>( in TEST() local
240 segment_label_sorter->GetNumItems()); in TEST()
[all …]
|
| /dports/misc/xgboost/xgboost-1.5.1/tests/cpp/objective/ |
| H A D | test_ranking_obj_gpu.cu | 135 auto segment_label_sorter = RankSegmentSorterTestImpl<float>( in TEST() local
166 std::vector<float> hgroup_dcgs(segment_label_sorter->GetNumGroups()); in TEST()
168 std::vector<uint32_t> hgroups(segment_label_sorter->GetNumGroups() + 1); in TEST()
170 EXPECT_EQ(hgroup_dcgs.size(), segment_label_sorter->GetNumGroups()); in TEST()
171 std::vector<float> hsorted_labels(segment_label_sorter->GetNumItems()); in TEST()
191 auto segment_label_sorter = RankSegmentSorterTestImpl<float>( in TEST() local
201 segment_label_sorter->CreateIndexableSortedPositions(); in TEST()
202 std::vector<uint32_t> sorted_indices(segment_label_sorter->GetNumItems()); in TEST()
218 auto segment_label_sorter = RankSegmentSorterTestImpl<float>( in TEST() local
240 segment_label_sorter->GetNumItems()); in TEST()
[all …]
|
| /dports/misc/py-xgboost/xgboost-1.5.1/src/objective/ |
| H A D | rank_obj.cu | 145 : dsorted_labels_(segment_label_sorter.GetItemsSpan()), in BaseLambdaWeightMultiplier()
146 dorig_pos_(segment_label_sorter.GetOriginalPositionsSpan()), in BaseLambdaWeightMultiplier()
147 dgroups_(segment_label_sorter.GetGroupsSpan()), in BaseLambdaWeightMultiplier()
275 : IndexablePredictionSorter(dpreds, segment_label_sorter), in NDCGLambdaWeightComputer()
276 dgroup_dcg_(segment_label_sorter.GetNumGroups(), 0.0f), in NDCGLambdaWeightComputer()
277 weight_multiplier_(segment_label_sorter, *this) { in NDCGLambdaWeightComputer()
511 : IndexablePredictionSorter(dpreds, segment_label_sorter), in MAPLambdaWeightComputer()
513 weight_multiplier_(segment_label_sorter, *this) { in MAPLambdaWeightComputer()
514 this->CreateMAPStats(dlabels, segment_label_sorter); in MAPLambdaWeightComputer()
526 auto nitems = segment_label_sorter.GetNumItems(); in CreateMAPStats()
[all …]
|
| /dports/misc/xgboost/xgboost-1.5.1/src/objective/ |
| H A D | rank_obj.cu | 145 : dsorted_labels_(segment_label_sorter.GetItemsSpan()), in BaseLambdaWeightMultiplier()
146 dorig_pos_(segment_label_sorter.GetOriginalPositionsSpan()), in BaseLambdaWeightMultiplier()
147 dgroups_(segment_label_sorter.GetGroupsSpan()), in BaseLambdaWeightMultiplier()
275 : IndexablePredictionSorter(dpreds, segment_label_sorter), in NDCGLambdaWeightComputer()
276 dgroup_dcg_(segment_label_sorter.GetNumGroups(), 0.0f), in NDCGLambdaWeightComputer()
277 weight_multiplier_(segment_label_sorter, *this) { in NDCGLambdaWeightComputer()
511 : IndexablePredictionSorter(dpreds, segment_label_sorter), in MAPLambdaWeightComputer()
513 weight_multiplier_(segment_label_sorter, *this) { in MAPLambdaWeightComputer()
514 this->CreateMAPStats(dlabels, segment_label_sorter); in MAPLambdaWeightComputer()
526 auto nitems = segment_label_sorter.GetNumItems(); in CreateMAPStats()
[all …]
|
| /dports/misc/py-xgboost/xgboost-1.5.1/src/metric/ |
| H A D | rank_metric.cu | 169 dh::SegmentSorter<float> segment_label_sorter; in EvalMetric() local
170 segment_label_sorter.SortItems(dlabels, pred_sorter.GetNumItems(), in EvalMetric()
176 ComputeDCG(pred_sorter, dlabels, ecfg, segment_label_sorter.GetOriginalPositionsSpan(), &idcg); in EvalMetric()
|
| /dports/misc/xgboost/xgboost-1.5.1/src/metric/ |
| H A D | rank_metric.cu | 169 dh::SegmentSorter<float> segment_label_sorter; in EvalMetric() local
170 segment_label_sorter.SortItems(dlabels, pred_sorter.GetNumItems(), in EvalMetric()
176 ComputeDCG(pred_sorter, dlabels, ecfg, segment_label_sorter.GetOriginalPositionsSpan(), &idcg); in EvalMetric()
|