Home
last modified time | relevance | path

Searched refs:aft_param_json (Results 1 – 4 of 4) sorted by relevance

/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/metric/
H A Dtest_survival_metric.cu75 auto aft_param_json = j_obj["aft_loss_param"]; in TEST() local
76 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution"]), "normal"); in TEST()
77 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution_scale"]), "10"); in TEST()
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/metric/
H A Dtest_survival_metric.cu75 auto aft_param_json = j_obj["aft_loss_param"]; in TEST() local
76 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution"]), "normal"); in TEST()
77 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution_scale"]), "10"); in TEST()
/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/objective/
H A Dtest_aft_obj.cc28 auto aft_param_json = j_obj["aft_loss_param"]; in TEST() local
29 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution"]), "logistic"); in TEST()
30 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution_scale"]), "5"); in TEST()
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/objective/
H A Dtest_aft_obj.cc28 auto aft_param_json = j_obj["aft_loss_param"]; in TEST() local
29 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution"]), "logistic"); in TEST()
30 EXPECT_EQ(get<String>(aft_param_json["aft_loss_distribution_scale"]), "5"); in TEST()