R/get-SpamTube-dataset.R

In christophM/interpretable-ml-book: Interpretable machine learning

get.ycomments.data = function(data_dir){
  ycomments.file = sprintf('%s/TubeSpam.csv', data_dir)
  if (!file.exists(ycomments.file)) {
    download.spam.data()
  }
  read.csv(ycomments.file, stringsAsFactors = FALSE)
}



# Download the youtube datasets 
download.spam.data = function(){
  urls = sprintf('http://lasid.sor.ufscar.br/labeling/datasets/%i/download/', 9:13)
  ycomments = lapply(urls, read.csv, stringsAsFactors=FALSE)
  ycomments = do.call('rbind', ycomments)
  cleanFun <- function(htmlString) {
    return(gsub("<.*?>", "", htmlString))
  }
  ycomments$CONTENT = cleanFun(ycomments$CONTENT)
  # Convert to ASCII
  ycomments$CONTENT = iconv(ycomments$CONTENT, "UTF-8", "ASCII", sub="")
  write.csv( x = ycomments, file = sprintf('%s/TubeSpam.csv', data_dir),row.names=FALSE)
}