classComp: Predict party membership with caret

In tjpajala/vaalikone: Scripts for analysing Finnish Election Machine datasets

Description

Perform k-fold CV and hyperparameter optimization with caret to predict party affiliation based on election machine questions.

Usage

res <- classComp(data, 10, 1, "rf", "party")

Arguments

data	The data set to be used: should only include party column and predictors!
k	Selected k for cross-validation.
model	The caret model to be used in prediction.
party_col	Column name that includes party affiliation.
repeats	Number of repeats for CV (increases stability).

Value

Data frame that includes optimized parameters, accuracy and kappa information.

tjpajala/vaalikone documentation built on May 26, 2019, 9:34 a.m.