我尝试使用 EJML wiki 中的示例,我们在这里使用 Levenberg Marquardt 优化源代码
我将它与这个.Net 版本进行比较,在该版本中,我们可以提供函数模型参数。
例如:a*x^2+b*x+c
我们可以将模型的所有参数作为输入。
但是,对于 EJML LM 代码,我看不到在哪里可以提供这些模型参数。
我在下面粘贴了我如何使用 LM EJML 类:
public class Main {
public static void main(String[] args) {
LevenbergMarquardt lm = new LevenbergMarquardt(new LevenbergMarquardt.Function() {
@Override
public void compute(DenseMatrix64F param, DenseMatrix64F x, DenseMatrix64F y) {
// TODO Auto-generated method stub
System.out.println("param:");
param.print();
System.out.println("X:");
x.print();
//y=a*x^2+b*x+c
for (int i = 0; i < x.numRows; i++) {
double xx = x.get(i, 0);
y.set(i, 0, param.get(0, 0) * xx * xx +
param.get(1, 0) * xx + param.get(2, 0));
}
System.out.println("Y:");
y.print();
}
});
//Seed inital parameters
lm.optimize(new DenseMatrix64F(new double[][]{{1}, {1}, {1}}),
new DenseMatrix64F(new double[][]{{0.1975}, {0.5084}, {0.7353}, {0.9706},
{1.1891}}), new DenseMatrix64F(new double[][]{{-0.0126}, {0.2311},
{0.4412}, {1.0210}, {1.1891}}));
}
}
那么我怎样才能给出这些模型参数呢?