我正在使用HR Analytics数据在R中实施逻辑回归(预测离开公司的概率:"如果为真,则离开" = 1,否则,基于两个特征为0。
对于我的最终模型,我有权重和训练数据。有2个功能,所以我的重量是2行矢量。
如何绘制决策边界?我确实在网上搜索,但到目前为止找不到解决方案。
### Load data
HR = read.csv(file = "HR_comma_sep.csv",header = TRUE, na.strings=c(""))
HR.2 <- HR[c("satisfaction_level","time_spend_company","left")]
### Split the data "HR.2" into training and testing data
data <- HR.2 [1:12000,]
test.data <- HR.2 [12001:14999,]
### Fit a logistic regression model using training data
model <- glm (left ~ satisfaction_level + time_spend_company,
family = binomial(link = 'logit'), data = data)
summary(model)
感谢您的帮助!