R/oolong_formats.R

In oolong: Create Validation Tests for Automated Content Analysis

.is_topic_model <- function(x) {
    if (any(class(x) %in% c("WarpLDA", "STM", "BTM", "keyATM_output","textmodel_lda", "textmodel_nb"))) {
        return(TRUE)
    }
    if (is.null(attr(class(x), "package"))) {
        return(FALSE)
    }
    if ("topicmodels" == attr(class(x), "package")) {
        return(TRUE)
    }
    return(FALSE)
}

##oolong for formats


.convert_input_model_s3 <- function(input_model) {
    if (!.is_topic_model(input_model)) {
        stop("input_model is not supported.")
    }
    output <- list()
    output$model <- input_model
    if ("WarpLDA" %in% class(input_model)) {
        class(output) <- append(class(output), "input_model_s3_warplda")
    } else if ("textmodel_lda" %in% class(input_model)) {
        class(output) <- append(class(output), "input_model_s3_seededlda")
    } else if ("textmodel_nb" %in% class(input_model)) {
        class(output) <- append(class(output), "input_model_s3_nb")
    } else if ("STM" %in% class(input_model)) {
        class(output) <- append(class(output), "input_model_s3_stm")
    } else if ("BTM" %in% class(input_model)) {
        class(output) <- append(class(output), "input_model_s3_btm")
    } else if ("keyATM_output" %in% class(input_model)) {
        class(output) <- append(class(output), "input_model_s3_keyatm")
    } else if ("topicmodels" == attr(class(input_model), "package")) {
        class(output) <- append(class(output), "input_model_s3_topicmodels")
    }    
    return(output)
}