R/create_corpus.R

In idiolect: Forensic Authorship Analysis

#' Create a corpus
#'
#' Function to read in text data and turn it into a `quanteda` corpus object.
#'
#' @param path A string containing the path to a folder of plain text files (ending in .txt) with their name structured as following: authorname_textname.txt (e.g. smith_text1.txt).
#'
#' @return A `quanteda` corpus object with the authors' names as a docvar.
#'
#' @examples
#' \dontrun{
#' path <- "path/to/data"
#' create_corpus(path)
#' }
#'
#' @export
create_corpus <- function(path){

  corpus <- readtext::readtext(file = paste0(path, "/*.txt"),
                               docvarsfrom = "filenames", docvarnames = c("author", "textname")) |>
    quanteda::corpus()

  return(corpus)

}