Transformer

CMakeLists.txt

@@ -22,6 +22,8 @@ add_library(neighborly_core
     src/OnlineSurrogate/OnlineSurrogate.h
     src/TrainingArchive/TrainingArchive.cpp
     src/TrainingArchive/TrainingArchive.h
+    src/TransformerRanker/TransformerRanker.cpp
+    src/TransformerRanker/TransformerRanker.h
     src/NeighborlyCore/plane.h
     src/NeighborlyCore/precise_geometry.h
     src/NeighborlyCore/util.cpp
@@ -61,6 +63,7 @@ target_include_directories(neighborly_core
         "${CMAKE_CURRENT_SOURCE_DIR}/src/SearchArchive"
         "${CMAKE_CURRENT_SOURCE_DIR}/src/OnlineSurrogate"
         "${CMAKE_CURRENT_SOURCE_DIR}/src/TrainingArchive"
+        "${CMAKE_CURRENT_SOURCE_DIR}/src/TransformerRanker"
         "${CMAKE_CURRENT_SOURCE_DIR}/projects/Szilassi"
         "${EIGEN_ROOT}"
 )
@@ -81,6 +84,14 @@ endif()
 add_executable(verify_cpp projects/VerifyCpp/verify_cpp.cpp)
 target_link_libraries(verify_cpp PRIVATE neighborly_core)
 
+add_executable(transformer_training_export
+    projects/TransformerTrainingExport/transformer_training_export.cpp
+)
+target_link_libraries(transformer_training_export PRIVATE neighborly_core)
+if(MSVC)
+    target_compile_options(transformer_training_export PRIVATE /utf-8)
+endif()
+
 add_executable(polyhedron_gui WIN32 projects/PolyhedronGui/launcher_gui.cpp)
 if(MSVC)
     target_compile_definitions(polyhedron_gui PRIVATE UNICODE _UNICODE NOMINMAX)
@@ -101,4 +112,15 @@ if(BUILD_TESTING)
     set_tests_properties(training_archive_selftest PROPERTIES
         WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}"
     )
+    add_executable(transformer_ranker_selftest
+        tests/TransformerRankerSelfTest/TransformerRankerSelfTest.cpp
+    )
+    target_link_libraries(transformer_ranker_selftest PRIVATE neighborly_core)
+    if(MSVC)
+        target_compile_options(transformer_ranker_selftest PRIVATE /utf-8)
+    endif()
+    add_test(NAME transformer_ranker_selftest COMMAND transformer_ranker_selftest)
+    set_tests_properties(transformer_ranker_selftest PROPERTIES
+        WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}"
+    )
 endif()