使用stepAIC()得到最终模型后如何提取变量名列表
R语言提取stepAIC最终模型变量名的方法
你之前使用attr(terms(backward.aic), "predvars")没有得到预期结果,是因为该属性返回的是语言类对象,而非可直接使用的字符型变量名列表。stepAIC返回的最终模型属于标准的lm/glm类对象,可通过以下两种常用方式提取你需要的变量列表:
提取原始输入变量名(包含分类变量的整体名称,不含因子拆分的水平项)
直接解析最终模型的公式即可,代码如下:# 提取全部变量,第1位为因变量,后续为自变量 all_vars <- all.vars(formula(backward.aic)) # 仅提取自变量列表 independent_vars <- all_vars[-1]如需同时保留因变量,直接使用
all_vars即可。提取模型系数对应的变量名(包含因子拆分的水平项、交互项,可自由选择是否保留截距)
直接提取模型系数的名称即可,代码如下:# 提取所有系数名,默认包含截距项"(Intercept)" all_coef_names <- names(backward.aic$coefficients) # 过滤掉截距项,仅保留预测变量对应的系数名 pred_coef_names <- all_coef_names[all_coef_names != "(Intercept)"]
如需核对提取结果是否正确,直接运行formula(backward.aic)打印最终模型公式即可对比验证。
内容的提问来源于stack exchange,提问作者jean10
相关产品推荐
相关产品推荐

