예제 #1
0
 def test_get_metric_name(self):
     model = CatBoostAlgorithm(self.params)
     self.assertEqual(model.get_metric_name(), "logloss")
     params = dict(self.params)
     params["loss_function"] = "MultiClass"
     model = CatBoostAlgorithm(params)
     self.assertEqual(model.get_metric_name(), "logloss")
예제 #2
0
 def test_get_metric_name(self):
     model = CatBoostAlgorithm(self.params)
     self.assertEqual(model.get_metric_name(), "rmse")