1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
#include "TH.h"
#include "luaT.h"
#ifdef _OPENMP
#include "omp.h"
#endif
#include "error.h"
#include "hash_map.h"
#define torch_(NAME) TH_CONCAT_3(torch_, Real, NAME)
#define torch_Tensor TH_CONCAT_STRING_3(torch., Real, Tensor)
#define nn_(NAME) TH_CONCAT_3(nn_, Real, NAME)
#include "generic/LogitBoostCriterion.c"
#include "THGenerateFloatTypes.h"
#include "generic/DFD.c"
#include "THGenerateFloatTypes.h"
#include "generic/S2D.c"
#include "THGenerateFloatTypes.h"
#include "generic/CartTree.c"
#include "THGenerateFloatTypes.h"
#include "GBDT_common.h"
#include "generic/GBDT.c"
#include "THGenerateFloatTypes.h"
static const struct luaL_Reg decisiontree_hash_map_routines[] = {
{"__gc", hash_map_gc_lua},
{"retain", hash_map_retain_lua},
{"metatablename", hash_map_metatablename_lua},
{"clear", hash_map_clear_lua},
{"put", hash_map_put_lua},
{"fill", hash_map_fill_lua},
{"adjustCounter", hash_map_adjust_counter_lua},
{"getCounter", hash_map_get_counter_lua},
{"setCounter", hash_map_set_counter_lua},
{"get", hash_map_get_lua},
{"getInplace", hash_map_get_inplace_lua},
{"del", hash_map_del_lua},
{"size", hash_map_size_lua},
{"safe", hash_map_autolock_on_lua},
{"unsafe", hash_map_autolock_off_lua},
{"toTensors", hash_map_to_tensor_lua},
{"new", hash_map_init_lua},
{NULL, NULL}
};
DLL_EXPORT int luaopen_libdecisiontree(lua_State *L)
{
// HashMap
luaL_newmetatable(L, "dt.HashMap");
lua_pushstring(L, "__index");
lua_pushvalue(L, -2);
lua_settable(L, -3);
luaT_setfuncs(L, decisiontree_hash_map_routines, 0);
nn_FloatLogitBoostCriterion_init(L);
nn_DoubleLogitBoostCriterion_init(L);
nn_FloatDFD_init(L);
nn_DoubleDFD_init(L);
nn_FloatS2D_init(L);
nn_DoubleS2D_init(L);
nn_FloatCT_init(L);
nn_DoubleCT_init(L);
nn_FloatGBDT_init(L);
nn_DoubleGBDT_init(L);
return 1;
}
|
