text2emotion (version 0.1.0)

evaluate_rf_model: 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
)

Value

A list with test accuracy, test predictions, and aligned test data.

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.