meta.h 1.36 KB
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

namespace LightGBM {

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

18
const score_t kMinScore = -std::numeric_limits<score_t>::infinity();
Guolin Ke's avatar
Guolin Ke committed
19

20
const score_t kEpsilon = 1e-15f;
Guolin Ke's avatar
Guolin Ke committed
21

Guolin Ke's avatar
Guolin Ke committed
22
const double kZeroThreshold = 1e-35f;
Guolin Ke's avatar
Guolin Ke committed
23

Guolin Ke's avatar
Guolin Ke committed
24
25

typedef int32_t comm_size_t;
Guolin Ke's avatar
Guolin Ke committed
26

Guolin Ke's avatar
Guolin Ke committed
27
using PredictFunction =
Guolin Ke's avatar
Guolin Ke committed
28
std::function<void(const std::vector<std::pair<int, double>>&, double* output)>;
Guolin Ke's avatar
Guolin Ke committed
29

Guolin Ke's avatar
Guolin Ke committed
30
31
32
33
34
35
typedef void(*ReduceFunction)(const char* input, char* output, int type_size, comm_size_t array_size);


typedef void(*ReduceScatterFunction)(char* input, comm_size_t input_size, int type_size,
                                     const comm_size_t* block_start, const comm_size_t* block_len, int num_block, char* output, comm_size_t output_size,
                                     const ReduceFunction& reducer);
ww's avatar
ww committed
36

Guolin Ke's avatar
Guolin Ke committed
37
38
typedef void(*AllgatherFunction)(char* input, comm_size_t input_size, const comm_size_t* block_start,
                                 const comm_size_t* block_len, int num_block, char* output, comm_size_t output_size);
ww's avatar
ww committed
39
40


Guolin Ke's avatar
Guolin Ke committed
41
42
#define NO_SPECIFIC (-1)

43
44
45
46
#if (_MSC_VER <= 1800)
#define __func__ __FUNCTION__
#endif

Guolin Ke's avatar
Guolin Ke committed
47
48
}  // namespace LightGBM

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