evaluate_rf_model {text2emotion} | R Documentation |
Evaluate a Random Forest Model on Test Data
Description
Evaluate a Random Forest Model on Test Data
Usage
evaluate_rf_model(
rf_model,
test_texts,
test_labels,
tfidf_model,
vectorizer,
stopwords,
verbose = TRUE
)
Arguments
rf_model |
A trained 'ranger' model object. |
test_texts |
A vector of raw test texts. |
test_labels |
A factor vector of true labels. |
tfidf_model |
The TF-IDF transformer used for training. |
vectorizer |
The vectorizer used to build DTM. |
stopwords |
A character vector of stopwords. |
verbose |
Whether to print progress. Default TRUE. |
Value
A list with test accuracy, test predictions, and aligned test data.
[Package text2emotion version 0.1.0 Index]