README.rst

linear-regression

This app creates and fits an Elastic Net model based on parquet-formatted input data. The arguments to the program are asfollows:

train: (str, required) local path or URI of a parquet file containing training data
test: (str, required) local path or URI of a parquet file containing test data
alpha: (float) alpha for the regressor; default .001
l1-ratio: (float) l1 ratio to be used for the regressor; default .5
label-col: (str, required) name of label column in dataset; string input
feat-cols: (str) names of columns in dataset to be used as features; input is one string with names delimited by commas. If no argument is provided, it is assumed that all columns but the label column are feature columns.

This app currently assumes that the input data is all numerical.

To run the app with default parameters while in the root directory, run the command

mlflow run apps/linear-regression -P train="insert/data/path/" -P test="insert/data/path/" -P label-col="insert.label.col"