# NOT RUN {
# ----- Example with objective and gradient written in R -----
set.seed(1234)
p <- 5; n <- 150
B <- matrix(rnorm(n*n), nrow=n)
B <- B + t(B)
D <- diag(p:1, p)
tx <- function(x) { matrix(x, n, p) }
f <- function(x) { X <- tx(x); Trace( t(X) %*% B %*% X %*% D ) }
g <- function(x) { X <- tx(x); 2 * B %*% X %*% D }
mod <- Module("ManifoldOptim_module", PACKAGE = "ManifoldOptim")
prob <- new(mod$RProblem, f, g)
x0 <- as.numeric(orthonorm(matrix(rnorm(n*p), nrow=n, ncol=p)))
mani.params <- get.manifold.params(IsCheckParams = TRUE)
solver.params <- get.solver.params(IsCheckParams = TRUE)
mani.defn <- get.stiefel.defn(n, p)
res <- manifold.optim(prob, mani.defn, method = "RTRSR1",
mani.params = mani.params, solver.params = solver.params, x0 = x0)
print(res)
head(tx(res$xopt))
# }
# NOT RUN {
library(ManifoldOptim)
library(RcppArmadillo)
# ----- Example with objective and gradient written in C++ -----
set.seed(1234)
p <- 5; n <- 150
B <- matrix(rnorm(n*n), nrow=n)
B <- B + t(B) # force symmetric
D <- diag(p:1, p)
# The Problem class is written in C++. Get a handle to it and set it up from R
Rcpp::sourceCpp(code = '
//[[Rcpp::depends(RcppArmadillo,ManifoldOptim)]]
#include <RcppArmadillo.h>
#include <ManifoldOptim.h>
using namespace Rcpp;
using namespace arma;
class BrockettProblem : public MatrixManifoldOptimProblem
{
public:
BrockettProblem(const arma::mat& B, const arma::mat& D)
: MatrixManifoldOptimProblem(false, true), m_B(B), m_D(D) { }
virtual ~BrockettProblem() { }
double objFun(const arma::mat& X) const {
return arma::trace(X.t() * m_B * X * m_D);
}
arma::mat gradFun(const arma::mat& X) const {
return 2 * m_B * X * m_D;
}
const arma::mat& GetB() const { return m_B; }
const arma::mat& GetD() const { return m_D; }
private:
arma::mat m_B;
arma::mat m_D;
};
RCPP_MODULE(Brockett_module) {
class_<BrockettProblem>("BrockettProblem")
.constructor<mat,mat>()
.method("objFun", &BrockettProblem::objFun)
.method("gradFun", &BrockettProblem::gradFun)
.method("GetB", &BrockettProblem::GetB)
.method("GetD", &BrockettProblem::GetD)
;
}
')
prob <- new(BrockettProblem, B, D)
X0 <- orthonorm(matrix(rnorm(n*p), nrow=n, ncol=p))
x0 <- as.numeric(X0)
tx <- function(x) { matrix(x, n, p) }
mani.params <- get.manifold.params(IsCheckParams = TRUE)
solver.params <- get.solver.params(DEBUG = 0, Tolerance = 1e-4,
Max_Iteration = 1000, IsCheckParams = TRUE, IsCheckGradHess = FALSE)
mani.defn <- get.stiefel.defn(n, p)
res <- manifold.optim(prob, mani.defn, method = "RTRSR1",
mani.params = mani.params, solver.params = solver.params, x0 = x0)
print(res)
head(tx(res$xopt))
# }
# NOT RUN {
# }
Run the code above in your browser using DataLab