[mlpack-git] master: Build the convolutional neural network code. (42d61df)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Thu Sep 3 08:35:45 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/91ae1062772a0f2bbca9a072769629c2d775ae64...42d61dfdbc9b0cbce59398e67ea58544b0fa1919
>---------------------------------------------------------------
commit 42d61dfdbc9b0cbce59398e67ea58544b0fa1919
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date: Thu Sep 3 14:35:22 2015 +0200
Build the convolutional neural network code.
>---------------------------------------------------------------
42d61dfdbc9b0cbce59398e67ea58544b0fa1919
src/mlpack/methods/ann/CMakeLists.txt | 3 +++
.../{performance_functions => convolution_rules}/CMakeLists.txt | 7 ++++---
src/mlpack/methods/ann/layer/CMakeLists.txt | 2 ++
.../initialization_methods => ann/pooling_rules}/CMakeLists.txt | 4 ++--
src/mlpack/tests/CMakeLists.txt | 1 +
5 files changed, 12 insertions(+), 5 deletions(-)
diff --git a/src/mlpack/methods/ann/CMakeLists.txt b/src/mlpack/methods/ann/CMakeLists.txt
index 8389fd7..1fb782c 100644
--- a/src/mlpack/methods/ann/CMakeLists.txt
+++ b/src/mlpack/methods/ann/CMakeLists.txt
@@ -2,6 +2,7 @@
# Anything not in this list will not be compiled into MLPACK.
set(SOURCES
ffn.hpp
+ cnn.hpp
network_traits.hpp
)
@@ -20,3 +21,5 @@ add_subdirectory(layer)
add_subdirectory(optimizer)
add_subdirectory(performance_functions)
add_subdirectory(trainer)
+add_subdirectory(pooling_rules)
+add_subdirectory(convolution_rules)
diff --git a/src/mlpack/methods/ann/performance_functions/CMakeLists.txt b/src/mlpack/methods/ann/convolution_rules/CMakeLists.txt
similarity index 82%
copy from src/mlpack/methods/ann/performance_functions/CMakeLists.txt
copy to src/mlpack/methods/ann/convolution_rules/CMakeLists.txt
index 5047f81..4511bef 100644
--- a/src/mlpack/methods/ann/performance_functions/CMakeLists.txt
+++ b/src/mlpack/methods/ann/convolution_rules/CMakeLists.txt
@@ -1,9 +1,10 @@
# Define the files we need to compile
# Anything not in this list will not be compiled into MLPACK.
set(SOURCES
- mse_function.hpp
- sse_function.hpp
- cee_function.hpp
+ border_modes.hpp
+ naive_convolution.hpp
+ fft_convolution.hpp
+ svd_convolution.hpp
)
# Add directory name to sources.
diff --git a/src/mlpack/methods/ann/layer/CMakeLists.txt b/src/mlpack/methods/ann/layer/CMakeLists.txt
index 82c63ae..ec51329 100644
--- a/src/mlpack/methods/ann/layer/CMakeLists.txt
+++ b/src/mlpack/methods/ann/layer/CMakeLists.txt
@@ -7,6 +7,8 @@ set(SOURCES
bias_layer.hpp
dropout_layer.hpp
linear_layer.hpp
+ conv_layer.hpp
+ pooling_layer.hpp
)
# Add directory name to sources.
diff --git a/src/mlpack/methods/perceptron/initialization_methods/CMakeLists.txt b/src/mlpack/methods/ann/pooling_rules/CMakeLists.txt
similarity index 91%
copy from src/mlpack/methods/perceptron/initialization_methods/CMakeLists.txt
copy to src/mlpack/methods/ann/pooling_rules/CMakeLists.txt
index d5d9c31..70ed3eb 100644
--- a/src/mlpack/methods/perceptron/initialization_methods/CMakeLists.txt
+++ b/src/mlpack/methods/ann/pooling_rules/CMakeLists.txt
@@ -1,8 +1,8 @@
# Define the files we need to compile
# Anything not in this list will not be compiled into MLPACK.
set(SOURCES
- random_init.hpp
- zero_init.hpp
+ max_pooling.hpp
+ mean_pooling.hpp
)
# Add directory name to sources.
diff --git a/src/mlpack/tests/CMakeLists.txt b/src/mlpack/tests/CMakeLists.txt
index 60a1163..fa65a7d 100644
--- a/src/mlpack/tests/CMakeLists.txt
+++ b/src/mlpack/tests/CMakeLists.txt
@@ -10,6 +10,7 @@ add_executable(mlpack_test
aug_lagrangian_test.cpp
cf_test.cpp
cli_test.cpp
+ convolutional_network_test.cpp
cosine_tree_test.cpp
decision_stump_test.cpp
det_test.cpp
More information about the mlpack-git
mailing list