用python对已经建立的随机森林模型dec_rf,输入真实案例的x.tobacco为2.3,x.ldl为3,x.age为45,x.famhist为1值对是否患有心脏病进行预测
时间: 2023-06-25 14:07:12 浏览: 87
cohesivegenerate2d.rar_cohesivegenerate2d_cohesive单元_tobacco64s_
假设你的随机森林模型dec_rf已经建立好并保存在了文件dec_rf.pkl中,那么可以通过以下代码进行预测:
```python
import pickle
# 加载模型
with open('dec_rf.pkl', 'rb') as f:
dec_rf = pickle.load(f)
# 构造输入特征
x = {'tobacco': 2.3, 'ldl': 3, 'age': 45, 'famhist': 1}
# 预测输出
y_pred = dec_rf.predict([list(x.values())])[0]
if y_pred == 0:
print('该患者没有心脏病')
else:
print('该患者有心脏病')
```
解释一下代码:
1. 首先使用pickle模块加载已经保存好的随机森林模型dec_rf.pkl。
2. 接着构造输入特征x,其中键为特征名称,值为实际值。
3. 使用模型的predict方法对输入特征进行预测,得到输出y_pred。
4. 最后根据输出y_pred来判断该患者是否患有心脏病,输出结果。
阅读全文