tmp
This commit is contained in:
parent
ecc73dfe11
commit
adf998fb8b
@ -10,13 +10,14 @@ macro(add_example name switch)
|
|||||||
endif()
|
endif()
|
||||||
endmacro()
|
endmacro()
|
||||||
|
|
||||||
add_example(ex1 ON)
|
add_example(ex1 OFF)
|
||||||
add_example(ex2 ON)
|
add_example(ex2 OFF)
|
||||||
add_example(ex3 ON)
|
add_example(ex3 OFF)
|
||||||
add_example(ex4 ON)
|
add_example(ex4 OFF)
|
||||||
add_example(ex5 ON)
|
add_example(ex5 OFF)
|
||||||
add_example(ex6 ON)
|
add_example(ex6 OFF)
|
||||||
add_example(ex7 ON)
|
add_example(ex7 OFF)
|
||||||
add_example(ex8 ON)
|
add_example(ex8 OFF)
|
||||||
add_example(ex9 ON)
|
add_example(ex9 OFF)
|
||||||
add_example(ex10 ON)
|
add_example(ex10 OFF)
|
||||||
|
add_example(cfg_ex ON)
|
46
example/cfg_ex.cpp
Normal file
46
example/cfg_ex.cpp
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
/********************************************************
|
||||||
|
* ██████╗ ███████╗████████╗██╗
|
||||||
|
* ██╔════╝ ██╔════╝╚══██╔══╝██║
|
||||||
|
* ██║ ███╗███████╗ ██║ ██║
|
||||||
|
* ██║ ██║╚════██║ ██║ ██║
|
||||||
|
* ╚██████╔╝███████║ ██║ ███████╗
|
||||||
|
* ╚═════╝ ╚══════╝ ╚═╝ ╚══════╝
|
||||||
|
* Generic Scientific Template Library
|
||||||
|
*
|
||||||
|
* Copyright (c) 2022 Yi Zhang (yizhang-geo@zju.edu.cn)
|
||||||
|
*
|
||||||
|
* The GSTL is distributed under a dual licensing scheme. You can redistribute
|
||||||
|
* it and/or modify it under the terms of the GNU Lesser General Public
|
||||||
|
* License as published by the Free Software Foundation, either version 2
|
||||||
|
* of the License, or (at your option) any later version. You should have
|
||||||
|
* received a copy of the GNU Lesser General Public License (LGPL) along with
|
||||||
|
* this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*
|
||||||
|
* If the terms and conditions of the LGPL v.2. would prevent you from using
|
||||||
|
* the GSTL, please consider the option to obtain a commercial license for a
|
||||||
|
* fee. These licenses are offered by the GSTL's original author. As a rule,
|
||||||
|
* licenses are provided "as-is", unlimited in time for a one time fee. Please
|
||||||
|
* send corresponding requests to: yizhang-geo@zju.edu.cn. Please do not forget
|
||||||
|
* to include some description of your company and the realm of its activities.
|
||||||
|
* Also add information on how to contact you by electronic and paper mail.
|
||||||
|
******************************************************/
|
||||||
|
|
||||||
|
#include "../lib/optimization/cmn_grad.h"
|
||||||
|
|
||||||
|
using namespace gctl;
|
||||||
|
|
||||||
|
int main(int argc, char *argv[]) try
|
||||||
|
{
|
||||||
|
common_gradient cmg;
|
||||||
|
cmg.init(2, 2);
|
||||||
|
cmg.fill_model_gradient(0, _1d_array({1.0, 0.5}));
|
||||||
|
cmg.fill_model_gradient(1, _1d_array({-0.6, 1.0}));
|
||||||
|
|
||||||
|
_1d_array x = cmg.get_common_gradient(false);
|
||||||
|
std::cout << "x = (" << x[0] << ", " << x[1] << ")\n";
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
catch (const std::exception& e)
|
||||||
|
{
|
||||||
|
GCTL_ShowWhatError(e.what(), GCTL_ERROR_ERROR, 0, 0, 0);
|
||||||
|
}
|
@ -39,5 +39,6 @@
|
|||||||
#include "optimization/sgd.h"
|
#include "optimization/sgd.h"
|
||||||
#include "optimization/gradnorm.h"
|
#include "optimization/gradnorm.h"
|
||||||
#include "optimization/dwa.h"
|
#include "optimization/dwa.h"
|
||||||
|
#include "optimization/cmn_grad.h"
|
||||||
|
|
||||||
#endif // _GCTL_OPTIMIZATION_H
|
#endif // _GCTL_OPTIMIZATION_H
|
@ -28,6 +28,7 @@
|
|||||||
#ifndef _GCTL_CLCG_H
|
#ifndef _GCTL_CLCG_H
|
||||||
#define _GCTL_CLCG_H
|
#define _GCTL_CLCG_H
|
||||||
|
|
||||||
|
#include "gctl/utility.h"
|
||||||
#include "gctl/core.h"
|
#include "gctl/core.h"
|
||||||
#include "gctl/maths.h"
|
#include "gctl/maths.h"
|
||||||
#include "gctl/algorithms.h"
|
#include "gctl/algorithms.h"
|
||||||
|
110
lib/optimization/cmn_grad.cpp
Normal file
110
lib/optimization/cmn_grad.cpp
Normal file
@ -0,0 +1,110 @@
|
|||||||
|
/********************************************************
|
||||||
|
* ██████╗ ██████╗████████╗██╗
|
||||||
|
* ██╔════╝ ██╔════╝╚══██╔══╝██║
|
||||||
|
* ██║ ███╗██║ ██║ ██║
|
||||||
|
* ██║ ██║██║ ██║ ██║
|
||||||
|
* ╚██████╔╝╚██████╗ ██║ ███████╗
|
||||||
|
* ╚═════╝ ╚═════╝ ╚═╝ ╚══════╝
|
||||||
|
* Geophysical Computational Tools & Library (GCTL)
|
||||||
|
*
|
||||||
|
* Copyright (c) 2023 Yi Zhang (yizhang-geo@zju.edu.cn)
|
||||||
|
*
|
||||||
|
* GCTL is distributed under a dual licensing scheme. You can redistribute
|
||||||
|
* it and/or modify it under the terms of the GNU Lesser General Public
|
||||||
|
* License as published by the Free Software Foundation, either version 2
|
||||||
|
* of the License, or (at your option) any later version. You should have
|
||||||
|
* received a copy of the GNU Lesser General Public License along with this
|
||||||
|
* program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*
|
||||||
|
* If the terms and conditions of the LGPL v.2. would prevent you from using
|
||||||
|
* the GCTL, please consider the option to obtain a commercial license for a
|
||||||
|
* fee. These licenses are offered by the GCTL's original author. As a rule,
|
||||||
|
* licenses are provided "as-is", unlimited in time for a one time fee. Please
|
||||||
|
* send corresponding requests to: yizhang-geo@zju.edu.cn. Please do not forget
|
||||||
|
* to include some description of your company and the realm of its activities.
|
||||||
|
* Also add information on how to contact you by electronic and paper mail.
|
||||||
|
******************************************************/
|
||||||
|
|
||||||
|
#include "cmn_grad.h"
|
||||||
|
|
||||||
|
gctl::common_gradient::common_gradient()
|
||||||
|
{
|
||||||
|
set_lcg_message(LCG_THROW);
|
||||||
|
}
|
||||||
|
|
||||||
|
gctl::common_gradient::common_gradient(size_t Ln, size_t Mn)
|
||||||
|
{
|
||||||
|
init(Ln, Mn);
|
||||||
|
}
|
||||||
|
|
||||||
|
gctl::common_gradient::~common_gradient(){}
|
||||||
|
|
||||||
|
void gctl::common_gradient::LCG_Ax(const array<double> &x, array<double> &ax)
|
||||||
|
{
|
||||||
|
matvec(t_, G_, x, NoTrans);
|
||||||
|
matvec(ax, G_, t_, Trans);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
void gctl::common_gradient::set_solver(const lcg_para ¶)
|
||||||
|
{
|
||||||
|
set_lcg_para(para);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
void gctl::common_gradient::init(size_t Ln, size_t Mn)
|
||||||
|
{
|
||||||
|
Ln_ = Ln;
|
||||||
|
Mn_ = Mn;
|
||||||
|
g_.resize(Mn_);
|
||||||
|
B_.resize(Mn_);
|
||||||
|
G_.resize(Ln_, Mn_);
|
||||||
|
t_.resize(Ln_);
|
||||||
|
gm_.resize(Ln_);
|
||||||
|
x_.resize(Ln_, 1.0);
|
||||||
|
filled_.resize(Ln_, false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
void gctl::common_gradient::fill_model_gradient(size_t id, const _1d_array &g)
|
||||||
|
{
|
||||||
|
if (id >= Ln_) throw std::runtime_error("[gctl::common_gradient] Invalid index.");
|
||||||
|
if (g.size() != Mn_) throw std::runtime_error("[gctl::common_gradient] Invalid array size.");
|
||||||
|
|
||||||
|
G_.fill_row(id, g);
|
||||||
|
filled_[id] = true;
|
||||||
|
gm_[id] = g.module();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const gctl::_1d_array &gctl::common_gradient::get_common_gradient(bool normalized)
|
||||||
|
{
|
||||||
|
for (size_t i = 0; i < Ln_; i++)
|
||||||
|
{
|
||||||
|
if (!filled_[i]) throw std::runtime_error("[gctl::common_gradient] Unfilled model gradient.");
|
||||||
|
}
|
||||||
|
filled_.assign(false);
|
||||||
|
|
||||||
|
G_.normalize(RowMajor);
|
||||||
|
matvec(B_, G_, x_, Trans);
|
||||||
|
|
||||||
|
g_.random_float(-1.0, 1.0, RdUniform);
|
||||||
|
g_.normalize();
|
||||||
|
LCG_Minimize(g_, B_, LCG_CG);
|
||||||
|
|
||||||
|
if (normalized) g_.normalize();
|
||||||
|
else
|
||||||
|
{
|
||||||
|
g_.normalize();
|
||||||
|
matvec(t_, G_, g_, NoTrans);
|
||||||
|
|
||||||
|
double tmod = t_.module(L1);
|
||||||
|
double gmod = 0.0;
|
||||||
|
for (size_t i = 0; i < Ln_; i++)
|
||||||
|
{
|
||||||
|
gmod += gm_[i]*abs(t_[i])/tmod;
|
||||||
|
}
|
||||||
|
g_.scale(gmod);
|
||||||
|
}
|
||||||
|
return g_;
|
||||||
|
}
|
92
lib/optimization/cmn_grad.h
Normal file
92
lib/optimization/cmn_grad.h
Normal file
@ -0,0 +1,92 @@
|
|||||||
|
/********************************************************
|
||||||
|
* ██████╗ ██████╗████████╗██╗
|
||||||
|
* ██╔════╝ ██╔════╝╚══██╔══╝██║
|
||||||
|
* ██║ ███╗██║ ██║ ██║
|
||||||
|
* ██║ ██║██║ ██║ ██║
|
||||||
|
* ╚██████╔╝╚██████╗ ██║ ███████╗
|
||||||
|
* ╚═════╝ ╚═════╝ ╚═╝ ╚══════╝
|
||||||
|
* Geophysical Computational Tools & Library (GCTL)
|
||||||
|
*
|
||||||
|
* Copyright (c) 2023 Yi Zhang (yizhang-geo@zju.edu.cn)
|
||||||
|
*
|
||||||
|
* GCTL is distributed under a dual licensing scheme. You can redistribute
|
||||||
|
* it and/or modify it under the terms of the GNU Lesser General Public
|
||||||
|
* License as published by the Free Software Foundation, either version 2
|
||||||
|
* of the License, or (at your option) any later version. You should have
|
||||||
|
* received a copy of the GNU Lesser General Public License along with this
|
||||||
|
* program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*
|
||||||
|
* If the terms and conditions of the LGPL v.2. would prevent you from using
|
||||||
|
* the GCTL, please consider the option to obtain a commercial license for a
|
||||||
|
* fee. These licenses are offered by the GCTL's original author. As a rule,
|
||||||
|
* licenses are provided "as-is", unlimited in time for a one time fee. Please
|
||||||
|
* send corresponding requests to: yizhang-geo@zju.edu.cn. Please do not forget
|
||||||
|
* to include some description of your company and the realm of its activities.
|
||||||
|
* Also add information on how to contact you by electronic and paper mail.
|
||||||
|
******************************************************/
|
||||||
|
|
||||||
|
#ifndef _GCTL_COMMON_GRADIENT_H
|
||||||
|
#define _GCTL_COMMON_GRADIENT_H
|
||||||
|
|
||||||
|
#include "lcg.h"
|
||||||
|
|
||||||
|
namespace gctl
|
||||||
|
{
|
||||||
|
class common_gradient : public lcg_solver
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
common_gradient(); ///< 构造函数
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Construct a new common_gradient object
|
||||||
|
*
|
||||||
|
* @param Ln Number of loss functions
|
||||||
|
* @param Mn Number of model parameters
|
||||||
|
*/
|
||||||
|
common_gradient(size_t Ln, size_t Mn);
|
||||||
|
|
||||||
|
virtual ~common_gradient(); ///< 析构函数
|
||||||
|
|
||||||
|
virtual void LCG_Ax(const array<double> &x, array<double> &ax); ///< 计算Ax
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Configure the solver's setups
|
||||||
|
*
|
||||||
|
* @param para LCG solver parameters
|
||||||
|
*/
|
||||||
|
void set_solver(const lcg_para ¶);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Initialize the common_gradient object
|
||||||
|
*
|
||||||
|
* @param Ln Number of loss functions
|
||||||
|
* @param Mn Number of model parameters
|
||||||
|
*/
|
||||||
|
void init(size_t Ln, size_t Mn);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Fill the model gradient
|
||||||
|
*
|
||||||
|
* @param id Loss function index
|
||||||
|
* @param g Model gradient
|
||||||
|
*/
|
||||||
|
void fill_model_gradient(size_t id, const _1d_array &g);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Get the conflict free gradient
|
||||||
|
*
|
||||||
|
* @param normalized Normalize the output gradient
|
||||||
|
* @return Calculated model gradient
|
||||||
|
*/
|
||||||
|
const _1d_array &get_common_gradient(bool normalized = true);
|
||||||
|
|
||||||
|
private:
|
||||||
|
size_t Ln_, Mn_; // Ln_: loss_func number,Mn_: model number
|
||||||
|
_2d_matrix G_;
|
||||||
|
_1d_array B_, g_, t_, x_;
|
||||||
|
_1d_array gm_;
|
||||||
|
array<bool> filled_;
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // _GCTL_COMMON_GRADIENT_H
|
@ -28,6 +28,7 @@
|
|||||||
#ifndef _GCTL_LBFGS_H
|
#ifndef _GCTL_LBFGS_H
|
||||||
#define _GCTL_LBFGS_H
|
#define _GCTL_LBFGS_H
|
||||||
|
|
||||||
|
#include "gctl/utility.h"
|
||||||
#include "gctl/core.h"
|
#include "gctl/core.h"
|
||||||
#include "gctl/maths.h"
|
#include "gctl/maths.h"
|
||||||
#include "gctl/algorithms.h"
|
#include "gctl/algorithms.h"
|
||||||
|
@ -28,6 +28,7 @@
|
|||||||
#ifndef _GCTL_SGD_H
|
#ifndef _GCTL_SGD_H
|
||||||
#define _GCTL_SGD_H
|
#define _GCTL_SGD_H
|
||||||
|
|
||||||
|
#include "gctl/utility.h"
|
||||||
#include "gctl/core.h"
|
#include "gctl/core.h"
|
||||||
#include "gctl/algorithms.h"
|
#include "gctl/algorithms.h"
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user