smile-kotlin/smile.validation/CrossValidationRunner/stratify

stratify

fun <T, M : Classifier<T>> stratify(k: Int, x: Array<T>, y: IntArray, trainer: (Array<T>, IntArray) -> M): ClassificationValidations<M>(source)

Stratified cross validation on a generic classifier.

Return

validation results.

Parameters

k-fold cross validation.

data samples.

sample labels.

trainer

a lambda to return a classifier trained on the given data.

fun <T, M : Classifier<T>> stratify(round: Int, k: Int, x: Array<T>, y: IntArray, trainer: (Array<T>, IntArray) -> M): ClassificationValidations<M>(source)

Repeated stratified cross validation on a generic classifier.

Return

validation results.

Parameters

round

the number of rounds of repeated cross validation.

k-fold cross validation.

data samples.

sample labels.

trainer

a lambda to return a classifier trained on the given data.

fun <M : DataFrameClassifier> stratify(k: Int, formula: Formula, data: DataFrame, trainer: (Formula, DataFrame) -> M): ClassificationValidations<M>(source)

Stratified cross validation on a data frame classifier.

Return

validation results.

Parameters

k-fold cross validation.

formula

model formula.

data

data samples.

trainer

a lambda to return a classifier trained on the given data.

fun <M : DataFrameClassifier> stratify(round: Int, k: Int, formula: Formula, data: DataFrame, trainer: (Formula, DataFrame) -> M): ClassificationValidations<M>(source)

Repeated stratified cross validation on a data frame classifier.

Return

validation results.

Parameters

round

the number of rounds of repeated cross validation.

k-fold cross validation.

formula

model formula.

data

data samples.

trainer

a lambda to return a classifier trained on the given data.