#include <mlpack/core.hpp>#include <mlpack/methods/xgboost/loss_functions/sse_loss.hpp>#include "catch.hpp"#include "serialization.hpp"
Include dependency graph for xgboost_test.cpp:
Functions | |
| TEST_CASE ("SSEInitialPredictionTest", "[XGBTest]") | |
| Test that the initial prediction is calculated correctly for SSE loss. | |
| TEST_CASE ("SSELeafValueTest", "[XGBTest]") | |
| Test that output leaf value is calculated correctly for SSE Loss. | |
| TEST_CASE ("SSEGainTest", "[XGBTest]") | |
| Test that the gain is computed correctly for SSE Loss. | |
Detailed Description
Tests for the XGBoost class and related classes.
mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.
