meta.h 690 Bytes
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
3
4
5
6
7
8
#ifndef LIGHTGBM_META_H_
#define LIGHTGBM_META_H_

#include <cstdint>

#include <limits>
#include <vector>
#include <functional>
Guolin Ke's avatar
Guolin Ke committed
9
#include <memory>
Guolin Ke's avatar
Guolin Ke committed
10
11
12
13
14
15

namespace LightGBM {

/*! \brief Type of data size, it is better to use signed type*/
typedef int32_t data_size_t;
/*! \brief Type of score, and gradients */
16
typedef float score_t;
Guolin Ke's avatar
Guolin Ke committed
17
18
19
20
21
22
23

const score_t kMinScore = -std::numeric_limits<score_t>::infinity();

const score_t kEpsilon = 1e-15f;

using ReduceFunction = std::function<void(const char*, char*, int)>;

Guolin Ke's avatar
Guolin Ke committed
24
25
26
27
28
using PredictFunction =
std::function<std::vector<double>(const std::vector<std::pair<int, double>>&)>;

#define NO_SPECIFIC (-1)

Guolin Ke's avatar
Guolin Ke committed
29
30
}  // namespace LightGBM

Guolin Ke's avatar
Guolin Ke committed
31
#endif   // LightGBM_META_H_