如何用回归模型输出单个预测值？OLS模型实操疑问

statsmodels的OLS模型predict()方法本身是为批量预测设计的，哪怕输入单个样本，也会返回一维数组。要得到单个值，只需要对预测结果做一步提取即可，同时要保证输入的特征格式和训练时的X一致。

具体步骤和代码示例：

1. 准备你要预测的单个体重值，比如target_weight = 600（替换成你的实际数值）
2. 按照模型训练时的特征计算逻辑，生成对应的特征值：

   x_feature = a * np.power(target_weight, b)
   

3. 给特征添加常数项，并且转换成二维数组（必须和训练时X的结构匹配，训练时X是sm.add_constant()生成的带常数项的二维数组）：

   x_pred = sm.add_constant([x_feature])
   

4. 执行预测并提取单个值：

   # 得到预测数组
   pred_array = result.predict(x_pred)
   # 提取单个值，两种方式都可以
   single_pred = pred_array[0]
   # 或者用item()方法更稳妥
   single_pred = pred_array.item()
   

这样就能得到你需要的单个预测脑质量值了。

内容的提问来源于stack exchange，提问作者Azazo8