如何绘制逻辑回归的决策边界 - RStudio

时间:2017-07-15 14:38:58

标签: r plot logistic-regression

我正在使用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) 

感谢您的帮助!

0 个答案:

没有答案