mlpack_linear_regression(1) | User Commands | mlpack_linear_regression(1) |
mlpack_linear_regression - simple linear regression and prediction
mlpack_linear_regression [-m unknown] [-l double] [-T string] [-t string] [-r string] [-V bool] [-M unknown] [-o string] [-h -v]
An implementation of simple linear regression and simple ridge regression using ordinary least squares. This solves the problem
where X (specified by '--training_file (-t)') and y (specified either as the last column of the input matrix '--training_file (-t)' or via the ’--training_responses_file (-r)' parameter) are known and b is the desired variable. If the covariance matrix (X'X) is not invertible, or if the solution is overdetermined, then specify a Tikhonov regularization constant (with '--lambda (-l)') greater than 0, which will regularize the covariance matrix to make it invertible. The calculated b may be saved with the ’--output_predictions_file (-o)' output parameter.
y = X * b + e
Optionally, the calculated value of b is used to predict the responses for another matrix X' (specified by the '--test_file (-T)' parameter):
and the predicted responses y' may be saved with the ’--output_predictions_file (-o)' output parameter. This type of regression is related to least-angle regression, which mlpack implements as the 'lars' program.
y' = X' * b
For example, to run a linear regression on the dataset 'X.csv' with responses ’y.csv', saving the trained model to 'lr_model.bin', the following command could be used:
$ linear_regression --training_file X.csv --training_responses_file y.csv --output_model_file lr_model.bin
Then, to use 'lr_model.bin' to predict responses for a test set 'X_test.csv', saving the predictions to 'X_test_responses.csv', the following command could be used:
$ linear_regression --input_model_file lr_model.bin --test_file X_test.csv --output_predictions_file X_test_responses.csv
For further information, including relevant papers, citations, and theory, consult the documentation found at http://www.mlpack.org or included with your distribution of mlpack.
18 November 2018 | mlpack-3.0.4 |