diff options
| author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2017-11-05 14:34:22 +0000 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2017-11-05 14:34:22 +0000 |
| commit | 33fed0759994c27f164d0b4ca6c420e7376bef83 (patch) | |
| tree | 9daf2ca52a96cfda8cccbf310d95e8851c364f4a /contrib/torch/torch7 | |
| parent | 111dfde1ad75526cecfcdd09d79bbd07454c8b0a (diff) | |
| download | rspamd-33fed0759994c27f164d0b4ca6c420e7376bef83.tar.gz rspamd-33fed0759994c27f164d0b4ca6c420e7376bef83.zip | |
[Feature] Add torch-decisiontree package
https://github.com/twitter/torch-decisiontree
This project implements random forests and gradient boosted decision
trees (GBDT). The latter uses gradient tree boosting. Both use ensemble
learning to produce ensembles of decision trees (that is, forests).
Diffstat (limited to 'contrib/torch/torch7')
| -rw-r--r-- | contrib/torch/torch7/cmake/TorchPackage.cmake | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/contrib/torch/torch7/cmake/TorchPackage.cmake b/contrib/torch/torch7/cmake/TorchPackage.cmake index 241dd195b..9ffdcf788 100644 --- a/contrib/torch/torch7/cmake/TorchPackage.cmake +++ b/contrib/torch/torch7/cmake/TorchPackage.cmake @@ -19,6 +19,10 @@ ENDMACRO() MACRO(ADD_TORCH_PACKAGE package src luasrc) INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR}) + INCLUDE_DIRECTORIES(${CMAKE_SOURCE_DIR}/contrib/torch/torch7/lib/TH) + INCLUDE_DIRECTORIES(${CMAKE_SOURCE_DIR}/contrib/torch/torch7/lib/luaT) + INCLUDE_DIRECTORIES(${CMAKE_BINARY_DIR}/contrib/torch/torch7/lib/TH) + INCLUDE_DIRECTORIES(${CMAKE_BINARY_DIR}/contrib/torch/torch7/lib/luaT) INCLUDE_DIRECTORIES(${Torch_LUA_INCLUDE_DIR}) ### C/C++ sources |