R/model_correct.R

In hongqi0314/PRAuto.PMML: User defined functions to output a PMML file that is to be implemented in S1

#' Correct vehicle model names
#'
#' Given a vehicle model name this function will produce the correct model name that
#' could be processed in S1.
#'
#' @param model Character string indicating vehicle model name
#'
#' @return Character string indicating corrected vehicle model name that is suited for S1
#'
#' @export
#'
#' @examples
#' model_correct('RAV4')
model_correct <- function(model) {
  model_s1 <- ''

  if (str_detect(model, regex('^fusion', ignore_case = TRUE))) {model_s1 <- 'Fusion'}
  else if (str_detect(model, regex('^mustang', ignore_case = TRUE))) {model_s1 <- 'Mustang'}
  else if (str_detect(model, regex('^accord', ignore_case = TRUE))) {model_s1 <- 'Accord'}
  else if (str_detect(model, regex('^accent', ignore_case = TRUE))) {model_s1 <- 'Accent'}
  else if (str_detect(model, regex('^elan', ignore_case = TRUE))) {model_s1 <- 'Elantra'}
  else if (str_detect(model, regex('^sonata', ignore_case = TRUE))) {model_s1 <- 'Sonata'}
  else if (str_detect(model, regex('^veloster', ignore_case = TRUE))) {model_s1 <- 'Veloster'}
  else if (str_detect(model, regex('^rio', ignore_case = TRUE))) {model_s1 <- 'Rio'}
  else if (str_detect(model, regex('^altima', ignore_case = TRUE))) {model_s1 <- 'Altima'}
  else if (str_detect(model, regex('^sentra', ignore_case = TRUE))) {model_s1 <- 'Sentra'}
  else if (str_detect(model, regex('^vers|^verz', ignore_case = TRUE))) {model_s1 <- 'Versa'}
  else if (str_detect(model, regex('^camry', ignore_case = TRUE))) {model_s1 <- 'Camry'}
  else if (str_detect(model, regex('^corol|^corl', ignore_case = TRUE))) {model_s1 <- 'Corolla'}
  else if (str_detect(model, regex('^tacoma', ignore_case = TRUE))) {model_s1 <- 'Tacoma'}
  else if (str_detect(model, regex('^yaris', ignore_case = TRUE))) {model_s1 <- 'Yaris'}
  else if (str_detect(model, regex('^rav', ignore_case = TRUE))) {model_s1 <- 'Rav4'}
  else {model_s1 <- model}

  return(model_s1)
}