SupervisedLearning
Supervised learning is the machine learning task of inferring a function from labeled training data. A supervised learning algorithm analyzes the training data and produces an inferred function, which can be used for mapping new examples.
Credits for the underlying algorithms of the functions that make predictions go to Andrew Ng at Stanford University.
Example
One example is the prediction of house prices (output value) along two dimensions (features): the size of house in square meters and the number of bedrooms.
The training data could look something like this:
| Size (m2) | # bedrooms | Price |
|---|---|---|
| 2104 | 3 | 399900 |
| 1600 | 3 | 329900 |
| 3000 | 4 | 539900 |
| 1940 | 4 | 239999 |
Using linear regression, we can now predict the price of a house with 2200 square meters and 3 bedrooms.
Installation
Add this line to your application's Gemfile:
gem 'supervised_learning'
And then execute:
$ bundle
Or install it yourself as:
$ gem install supervised_learning
Usage
- Create a matrix of the training data.
The output value (the type of value you want to predict) needs to be in the last column. The matrix has to have a) at least two columns (one feature and one output) and b) one row. The more data you feed it, the more accurate the prediction.
Consult the Ruby API for information on how to build matrices for instances based on arrays of rows or columns.
require 'matrix'
training_set = Matrix[ [2104, 3, 399900], [1600, 3, 329900], [3000, 4, 539900], [1940, 4, 239999] ]
- Instantiate an object with the training data.
program = SupervisedLearning::LinearRegression.new(training_set)
- Create a prediction in form of a matrix.
This matrix has one row and the columns follow the order of the training set. It has one column less than the training set since the output value (the last column of the training set) is the value we want to predict.
# Predict the price of a house of 2000 square meters with 3 bedrooms
prediction_set = Matrix[ [2000, 3] ]
program.predict(prediction_set)
=>
Contributing
- Fork it ( https://github.com/[my-github-username]/supervised_learning/fork )
- Create your feature branch (
git checkout -b my-new-feature) - Commit your changes (
git commit -am 'Add some feature') - Push to the branch (
git push origin my-new-feature) - Create a new Pull Request