Получить уровень факторов от обученного randomForest в R

Question

У меня есть обученная модель randomForest в R, которую я загружаю из файла RDS. Одним из предикторов является неупорядоченный фактор, у меня больше нет данных обучения, которые использовались для его получения, но хотелось бы знать, какие уровни имел этот фактор, чтобы я мог гарантировать, что данные, на которых я пытаюсь предсказать, имеют только эти уровнидля этого фактора. Возможно ли это?

Answer 1

Структура данных объекта randomForest немного варьируется в зависимости от того, обучен ли он с использованием «интерфейса формулы» или «интерфейса матрицы». Однако информация о независимых переменных хранится в обоих случаях в виде объекта списка randomForest$forest$xlevels.

library("randomForest")

df = read.csv("Audit.csv")

rf = randomForest(Adjusted ~ ., data = df)
print(rf$forest$xlevels)
print(rf$forest$xlevels["Education"])