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]