Add torch methods to discretize

Add fit_transform methods
This commit is contained in:
2024-06-07 23:54:42 +02:00
parent 633aa52849
commit e205668906
14 changed files with 183 additions and 87 deletions

View File

@@ -1,9 +1,9 @@
cmake_minimum_required(VERSION 3.20)
project(mdlp)
set(CMAKE_CXX_STANDARD 11)
add_library(mdlp CPPFImdlp.cpp Metrics.cpp BinDisc.cpp)
set(CMAKE_CXX_STANDARD 17)
find_package(Torch REQUIRED)
include_directories(${TORCH_INCLUDE_DIRS})
add_library(mdlp CPPFImdlp.cpp Metrics.cpp BinDisc.cpp Discretizer.cpp)
target_link_libraries(mdlp "${TORCH_LIBRARIES}")
add_subdirectory(sample)
add_subdirectory(tests)
add_subdirectory(tests)