mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-20 00:51:42 +00:00
[sysid] Add SysId (#5672)
The source is copied from this commit:
625ff04784.
This commit is contained in:
48
sysid/src/main/native/cpp/analysis/OLS.cpp
Normal file
48
sysid/src/main/native/cpp/analysis/OLS.cpp
Normal file
@@ -0,0 +1,48 @@
|
||||
// Copyright (c) FIRST and other WPILib contributors.
|
||||
// Open Source Software; you can modify and/or share it under the terms of
|
||||
// the WPILib BSD license file in the root directory of this project.
|
||||
|
||||
#include "sysid/analysis/OLS.h"
|
||||
|
||||
#include <tuple>
|
||||
#include <vector>
|
||||
|
||||
#include <Eigen/Cholesky>
|
||||
#include <Eigen/Core>
|
||||
|
||||
using namespace sysid;
|
||||
|
||||
std::tuple<std::vector<double>, double, double> sysid::OLS(
|
||||
const Eigen::MatrixXd& X, const Eigen::VectorXd& y) {
|
||||
assert(X.rows() == y.rows());
|
||||
|
||||
// The linear model can be written as follows:
|
||||
// y = Xβ + u, where y is the dependent observed variable, X is the matrix
|
||||
// of independent variables, β is a vector of coefficients, and u is a
|
||||
// vector of residuals.
|
||||
|
||||
// We want to minimize u² = uᵀu = (y - Xβ)ᵀ(y - Xβ).
|
||||
// β = (XᵀX)⁻¹Xᵀy
|
||||
|
||||
// Calculate β that minimizes uᵀu.
|
||||
Eigen::MatrixXd beta = (X.transpose() * X).llt().solve(X.transpose() * y);
|
||||
|
||||
// We will now calculate R² or the coefficient of determination, which
|
||||
// tells us how much of the total variation (variation in y) can be
|
||||
// explained by the regression model.
|
||||
|
||||
// We will first calculate the sum of the squares of the error, or the
|
||||
// variation in error (SSE).
|
||||
double SSE = (y - X * beta).squaredNorm();
|
||||
|
||||
int n = X.cols();
|
||||
|
||||
// Now we will calculate the total variation in y, known as SSTO.
|
||||
double SSTO = ((y.transpose() * y) - (1.0 / n) * (y.transpose() * y)).value();
|
||||
|
||||
double rSquared = (SSTO - SSE) / SSTO;
|
||||
double adjRSquared = 1.0 - (1.0 - rSquared) * ((n - 1.0) / (n - 3.0));
|
||||
double RMSE = std::sqrt(SSE / n);
|
||||
|
||||
return {{beta.data(), beta.data() + beta.rows()}, adjRSquared, RMSE};
|
||||
}
|
||||
Reference in New Issue
Block a user