regression

fun <T, M : Regression<T>> regression(x: Array<T>, y: DoubleArray, testx: Array<T>, testy: DoubleArray, trainer: (Array<T>, DoubleArray) -> M): RegressionValidation<M>(source)

Test a generic regression model on a train/validation split. The RMSE will be measured and printed out on standard output.

Return

the validation results.

Parameters

x

training data.

y

response variable of training data.

testx

test data.

testy

response variable of test data.

trainer

a lambda to return a regression model trained on the given data.


fun <M : DataFrameRegression> regression(formula: Formula, train: DataFrame, test: DataFrame, trainer: (Formula, DataFrame) -> M): RegressionValidation<M>(source)

Test a data frame regression model on a train/validation split. The RMSE will be measured and printed out on standard output.

Return

the validation results.

Parameters

formula

model formula.

train

training data.

test

test data.

trainer

a lambda to return a regression model trained on the given data.