9 lines
427 B
CMake
9 lines
427 B
CMake
include_directories(${PyWrap_SOURCE_DIR}/lib/Files)
|
|
include_directories(${PyWrap_SOURCE_DIR}/lib/json/include)
|
|
include_directories(${Python3_INCLUDE_DIRS})
|
|
include_directories(${TORCH_INCLUDE_DIRS})
|
|
|
|
add_executable(main main.cc STree.cc SVC.cc RandomForest.cc PyClassifier.cc PyWrap.cc)
|
|
|
|
target_link_libraries(main ${Python3_LIBRARIES} "${TORCH_LIBRARIES}" ${LIBTORCH_PYTHON} Boost::boost Boost::python Boost::numpy ArffFiles)
|