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")
def test_get_metric_name(self): model = CatBoostAlgorithm(self.params) self.assertEqual(model.get_metric_name(), "rmse")