#' Generating example data with mtcars.
#'
#' @description Generates test data using base R's mtcars dataset.
#' The response variable \code{y} is horsepower (hp), while the remaining variables
#' represent the predictive features \code{X}.
#'
#' @param seed random seed to use.
#' Defaults to 99.
#'
#' @return \code{X_train}, \code{y_train}, \code{X_val}, \code{y_val} (as a list of dataframes)
#'
#' @export
mtcars_data <- function(seed=99) {
y <- mtcars$hp
X <- mtcars
X$hp <- NULL
set.seed(seed)
training <- sample(rep(c(TRUE, TRUE, TRUE, FALSE), nrow(mtcars) / 4))
# Training Data
X_train <- X[training,]
y_train <- y[training]
# Validation Data
X_val <- X[!training,]
y_val <- y[!training]
return(list(X_train, y_train, X_val, y_val))
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.