1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
|
/* -------------------------------------------------------------------------- *
* Simbody(tm) Example: Unconstrained Optimization w/Analytical Gradient *
* -------------------------------------------------------------------------- *
* This is part of the SimTK biosimulation toolkit originating from *
* Simbios, the NIH National Center for Physics-Based Simulation of *
* Biological Structures at Stanford, funded under the NIH Roadmap for *
* Medical Research, grant U54 GM072970. See https://simtk.org/home/simbody. *
* *
* Portions copyright (c) 2006-12 Stanford University and the Authors. *
* Authors: Jack Middleton *
* Contributors: *
* *
* Licensed under the Apache License, Version 2.0 (the "License"); you may *
* not use this file except in compliance with the License. You may obtain a *
* copy of the License at http://www.apache.org/licenses/LICENSE-2.0. *
* *
* Unless required by applicable law or agreed to in writing, software *
* distributed under the License is distributed on an "AS IS" BASIS, *
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
* See the License for the specific language governing permissions and *
* limitations under the License. *
* -------------------------------------------------------------------------- */
#include "SimTKmath.h"
#include <iostream>
using namespace SimTK;
/* adapted from itkLBFGSOptimizerTest.cxx */
const static int NUMBER_OF_PARAMETERS = 2;
class ProblemSystem : public OptimizerSystem {
public:
ProblemSystem( int numParameters) : OptimizerSystem( numParameters){}
int objectiveFunc( const Vector &coefficients, bool new_coefficients, Real& f ) const override {
const Real x = coefficients[0];
const Real y = coefficients[1];
f = 0.5*(3*x*x+4*x*y+6*y*y) - 2*x + 8*y;
return(0);
}
int gradientFunc( const Vector &coefficients, bool new_coefficients, Vector &gradient )const override {
const Real x = coefficients[0];
const Real y = coefficients[1];
gradient[0] = 3*x + 2*y -2;
gradient[1] = 2*x + 6*y +8;
return(0);
}
};
int main() {
ProblemSystem sys(NUMBER_OF_PARAMETERS);
Vector results(NUMBER_OF_PARAMETERS);
Real f = NaN;
try {
Optimizer opt( sys );
opt.setConvergenceTolerance( .0001 );
results[0] = 100;
results[1] = -100;
f = opt.optimize( results );
}
catch (const SimTK::Exception::Base& e) {
std::cout << "UnconstrainedOptimization.cpp Caught exception :" << std::endl;
std::cout << e.what() << std::endl;
}
printf(" Optimal solution: f = %f parameters = %f %f \n",f,results[0],results[1]);
return 0;
}
|