conjugate gradient method tested

1 files changed, 10 insertions, 1 deletions

diff --git a/main.cpp b/main.cpp
index b479f59..09821d2 100644
--- a/main.cpp
+++ b/main.cpp
@@ -9,7 +9,7 @@
 
 INITIALIZE_EASYLOGGINGPP
 
-using Eigen::MatrixXd;
+using namespace Eigen;
 namespace po = boost::program_options;
 
 po::variables_map vm;
@@ -76,6 +76,15 @@ int main(int argc, char **argv) {
     po::store(po::parse_command_line(argc, argv, desc), vm);
     po::notify(vm);
 
+    /* Conjugate Gradient method test
+    MatrixXd A(3,3);
+    VectorXd b(3),x(3);
+    A<< 1,2,3,2,2,4,3,4,1;
+    b<< 1,1,1;
+    x<< 0,0,0;
+    cg_solve(A,b,x);
+    write_stream(std::cout,x);*/
+
     // Print help if necessary
     if (vm.count("help") || !(vm.count("train") || vm.count("validate") || vm.count("predict"))) {
         std::cout << desc;