wehub-resource-sync
38 lines
1.4 KiB
C++
38 lines
1.4 KiB
C++
/*!
|
|
* Copyright (c) 2017-2026 Microsoft Corporation. All rights reserved.
|
|
* Copyright (c) 2017-2026 The LightGBM developers. All rights reserved.
|
|
* Licensed under the MIT License. See LICENSE file in the project root for license information.
|
|
*/
|
|
#ifndef LIGHTGBM_INCLUDE_LIGHTGBM_PREDICTION_EARLY_STOP_H_
|
|
#define LIGHTGBM_INCLUDE_LIGHTGBM_PREDICTION_EARLY_STOP_H_
|
|
|
|
#include <LightGBM/export.h>
|
|
|
|
#include <string>
|
|
#include <functional>
|
|
|
|
namespace LightGBM {
|
|
|
|
struct PredictionEarlyStopInstance {
|
|
/// Callback function type for early stopping.
|
|
/// Takes current prediction and number of elements in prediction
|
|
/// @returns true if prediction should stop according to criterion
|
|
using FunctionType = std::function<bool(const double*, int)>;
|
|
|
|
FunctionType callback_function; // callback function itself
|
|
int round_period; // call callback_function every `runPeriod` iterations
|
|
};
|
|
|
|
struct PredictionEarlyStopConfig {
|
|
int round_period;
|
|
double margin_threshold;
|
|
};
|
|
|
|
/// Create an early stopping algorithm of type `type`, with given round_period and margin threshold
|
|
LIGHTGBM_EXPORT PredictionEarlyStopInstance CreatePredictionEarlyStopInstance(const std::string& type,
|
|
const PredictionEarlyStopConfig& config);
|
|
|
|
} // namespace LightGBM
|
|
|
|
#endif // LIGHTGBM_INCLUDE_LIGHTGBM_PREDICTION_EARLY_STOP_H_
|