CODEWITHSUNDEEP
rubylinear-regressionleast-squares
Peter C
Peter C

Reputation: 101

Ruby Library for doing Linear or NonLinear Least Squares Approximation?

Is there a Ruby library that allows me to do either linear or non-linear least squares approximation of a set of data.

What I would like to do is the following:

I'd prefer not to have to try to port some C/C++/Java library to get this functionality so I'm hoping there is some existing Ruby lib that I can use.

Upvotes: 10

Views: 2709

Answers (3)

Joachim W
Joachim W

Reputation: 8167

I am maintaining a C library for non-linear least squares minimization, http://apps.jcns.fz-juelich.de/lmfit, that comes with swig files for Ruby.

Upvotes: 1

dpott197
dpott197

Reputation: 1048

Try using the 'statsample' gem. You can perform logarithmic, exponential, power, or any other transformation using the example that is provided below. I hope this helps. 

require 'statsample'

# Independent Variable
x_data = [Math.exp(1), Math.exp(2), Math.exp(3), Math.exp(4), Math.exp(5)]

# Dependent Variable
y_data = [3, 5, 7, 9, 11]

# Logarithmic Transformation of X data 
# Math.log in Ruby has the base of Euler's number 'e' ~= '2.71828', 
# instead of the base '10'. Just a note.
log_x_data = x_data.map { |x| Math.log(x) }

# Linear Regression using the Logarithmic Transformation
x_vector=log_x_data.to_vector(:scale)
y_vector=y_data.to_vector(:scale)
ds={'x'=>x_vector,'y'=>y_vector}.to_dataset
mlr=Statsample::Regression.multiple(ds,'y')
mlr.summary

# Provides the value of the y-intercept 
#p mlr.constant

# Lists the coefficients of each casual variable. In this case, we have only one--'x'.
#p mlr.coeffs

# The regression output produces the line y = 1 + 2*x, but 
# considering that we transformed x earlier, it really produces
# y = 1 + 2*ln(x).

Upvotes: 8

Serabe
Serabe

Reputation: 3924

I used this snippet to work out some regressions. The first parameter is an array containing the x coordinates, the second an array containing the y coordinates and the last is the degree of the polynomial you are looking for. Not sure if it is this what you are looking for, but hopes it helps.

Upvotes: 7

Related Questions